Update Geo/GeochatP-main/app.py

Geo/GeochatP-main/app.py

@@ -14,7 +14,7 @@ class MyModel(torch.nn.Module):
         return x
 
 model = MyModel()
-model.load_state_dict(torch.load("model.pth"))
+model.load_state_dict(torch.load("model.pth", map_location=torch.device("cpu")))
 model.eval()
 
 # Define image preprocessing
@@ -31,5 +31,5 @@ def predict(image):
     return output.numpy().tolist()
 
 # Create Gradio interface
-iface = gr.Interface(fn=predict, inputs=gr.Image(), outputs="json")
+iface = gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs="json")
 iface.launch()