Update app.py

app.py

@@ -103,11 +103,26 @@ def sepia(input_img):
 
     custom_template = "my_custom_template.html"
 
+def explain_prediction(input_img):
+    # 모델 예측 수행
+    prediction = my_model.predict(input_img)
+
+    # 예측 결과를 설명하는 텍스트 생성
+    explanation = f"The model predicts this image as class {prediction}."
+
+    # 시각화: 예측 설명 텍스트를 이미지에 표시
+    result_image = input_img.copy()
+    cv2.putText(result_image, explanation, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
+
+    return result_image
+
+
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
                     title="SWJIN11 TASK",
                     description="경제학과 202211357 진성원",
+                    interpretation="default",
                     examples=["ADE_val_00000001.jpeg", "ADE_val_00001248.jpg", "image1.jpg", "image2.jpg", "image3.jpg", "image4.jpg"],
                     allow_flagging='never')