Update app.py

app.py

@@ -40,9 +40,11 @@ def predict(image):
         _, predicted = torch.max(outputs, 1)
     label = class_names[predicted.item()]
     print("Predicted label:", label)
-    return label
+    
+    # 예측 결과와 원본 출력값을 모두 반환
+    return label, outputs.numpy()  # 원본 출력값을 NumPy 배열로 변환하여 반환
 
-
-# Gradio 인터페이스 설정
-iface = gr.Interface(fn=predict, inputs=gr.Image(type='pil'), outputs='label', live=True)
+# Gradio 인터페이스 설정 (두 개의 출력)
+iface = gr.Interface(fn=predict, inputs=gr.Image(type='pil'), outputs=['label', 'numpy'], live=True)
 iface.launch()
+