Update app.py

app.py

@@ -108,12 +108,11 @@ def sepia(input_img):
     for class_name, prob in class_probabilities.items():
         print(f"{class_name}: {prob:.4f}")
 
-    # 정확성이 가장 높은 물체 정보 얻기
+    # 정확성이 가장 높은 물체 정보 출력
     max_prob_class = max(class_probabilities, key=class_probabilities.get)
     max_prob_value = class_probabilities[max_prob_class]
+    print(f"\nPredicted class with highest probability: {max_prob_class}, Probability: {max_prob_value:.4f}")
 
-    # 출력 및 반환
-    print(f"Predicted class with highest probability: {max_prob_class}, Probability: {max_prob_value:.4f}")
     return fig, f"Predicted class with highest probability: {max_prob_class}, Probability: {max_prob_value:.4f}"