Update app.py

app.py

@@ -28,7 +28,7 @@ class PretrainedModel(nn.Module):
 model = PretrainedModel(num_classes=19)
 
 # Carica i pesi con `weights_only=True` per evitare problemi di sicurezza
-state_dict = torch.load('model_v11.pt', map_location=torch.device('cpu'), weights_only=True)
+state_dict = torch.load('model_v11.pt', map_location=torch.device('cpu'))
 model.load_state_dict(state_dict)
 model.eval()  # Imposta il modello in modalità valutazione
 
@@ -50,11 +50,11 @@ def classify_image(img):
         
     return f"Class {predicted_class_index}, Confidence: {predicted_probability:.4f}"
 
-# Configura Gradio con l'API aggiornata
+# Configura Gradio per l'accesso pubblico
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil"),
     outputs="text"
 )
 
-iface.launch()
+iface.launch(share=True)  # Abilita l'accesso pubblico