changes

app.py

@@ -46,7 +46,7 @@ def predict(image):
 
     with torch.no_grad():
         outputs = model(img_tensor)
-        probabilities = torch.nn.functional.softmax(outputs[0], dim=0)
+        probabilities = torch.nn.functional.softmax(outputs.squeeze(0), dim=0)
     
     return {LABELS[i]: float(probabilities[i]) for i in range(len(LABELS))}
 
@@ -59,6 +59,4 @@ demo = gradio.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", 
-        server_port=7860, 
-        ssr_mode=False)
+    demo.launch()