Update app.py

app.py

@@ -1,9 +1,9 @@
 
-
+import torch
 import requests
 import gradio as gr
 from PIL import Image
-
+from transformers import ResNetForImageClassification, AutoImageProcessor
 
 target_folder = "Kang-Seong-Jun/Korean_Real_Estate_Classifier"
 
@@ -12,15 +12,6 @@ def load_model_and_preprocessor(target_folder):
     image_processor = AutoImageProcessor.from_pretrained(target_folder)
     return model, image_processor
 
-def fetch_image(url):
-    headers = {
-        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36'
-    }
-    image_raw = requests.get(url, headers=headers, stream=True).raw
-    image = Image.open(image_raw)
-
-    return image
-
 def infer_image(image, model, image_processor, k):
     processed_img = image_processor(images=image.convert("RGB"), return_tensors="pt")
 
@@ -36,21 +27,19 @@ def infer_image(image, model, image_processor, k):
         res += f"{idx+1}. {model.config.id2label[index.item()]:<15} ({prob.item()*100:.2f} %) \n" 
     return res
 
-def infer(url, k, target_folder=target_folder):
-    try : 
-        image = fetch_image(url)
+def infer(image, k, target_folder=target_folder):
+    try: 
         model, image_processor = load_model_and_preprocessor(target_folder)
-        res =  infer_image(image, model, image_processor, k)
-    except : 
+        res = infer_image(image, model, image_processor, k)
+    except Exception as e: 
         image = Image.new('RGB', (224, 224))
-        res = "이미지를 불러오는데 문제가 있나봐요. 다른 이미지 url로 다시 시도해주세요."
+        res = f"이미지를 처리하는데 문제가 발생했습니다: {str(e)}"
     return image, res 
 
 demo = gr.Interface(
     fn=infer,
     inputs=[
-        gr.Textbox(value="https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRpE-UHBp8ZufNUd3BKw8gtIxSe3IUwspOfqw&s", 
-                   label="Image URL"),
+        gr.Image(type="pil", label="입력 이미지"),
         gr.Slider(minimum=0, maximum=20, step=1, value=3, label="상위 몇개까지 보여줄까요?")
     ],
     outputs=[
@@ -59,4 +48,4 @@ demo = gr.Interface(
     ],
 )
 
-demo.launch()
+demo.launch()