Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import torch
 import open_clip
 
-def predict(image):
+def predict(input_image):
     device = torch.device("cpu")
     model, _, preprocess = open_clip.create_model_and_transforms('ViT-L-14', pretrained='openai', device=device)
 
-    image = preprocess().unsqueeze(0).to(device)
+    image = preprocess(input_image).unsqueeze(0).to(device)
 
     with torch.amp.autocast(device_type=device.type):
         with torch.no_grad():
@@ -20,5 +20,5 @@ def predict(image):
 
     return "Map" if result == 1 else "No map"
 
-demo = gr.Interface(fn=predict, inputs=gr.Image(label="Input Image", type="pil"), outputs="text")
+demo = gr.Interface(fn=predict, inputs=gr.Image(label="Check whether the image is a map or not", type="pil"), outputs="text")
 demo.launch()