Update app.py

app.py

@@ -1,4 +1,3 @@
-import argparse
 import json
 import time
 from PIL import Image
@@ -6,9 +5,6 @@ import torch
 from torchvision.transforms import transforms
 import gradio as gr
 
-parser = argparse.ArgumentParser(description="Image Classification")
-parser.add_argument("-i", "--image_path", required=True, help="Path to the image file")
-args = parser.parse_args()
 
 model = torch.load('model.pth', map_location=torch.device('cpu'))
 model.eval()
@@ -57,7 +53,7 @@ def create_tags(image):
 
 demo = gr.Interface(
     fn=create_tags,
-    inputs=["image"],
+    inputs=[gr.Image(type="pil")],
     outputs=["text"],
 )
 demo.launch()