Update app.py

app.py

@@ -1,3 +1,6 @@
+
+
+
 import gradio as gr
 from transformers import pipeline
 
@@ -8,10 +11,10 @@ def classify_image(image):
     predictions = model(image)
     return predictions
 
-# Gradio 인터페이스 생성 (수정된 부분)
+# Gradio 인터페이스 생성 (shape 인자 없이 수정)
 iface = gr.Interface(fn=classify_image,
-                     inputs=gr.Image(shape=(224, 224)),
-                     outputs=gr.Label(num_top_classes=3),
+                     inputs=gr.inputs.Image(),
+                     outputs=gr.outputs.Label(num_top_classes=3),
                      title="이미지 분류기",
                      description="이미지를 업로드하면, 사물을 인식하고 설명을 출력합니다.")