Update app.py

app.py

@@ -27,7 +27,7 @@ def classify_image(input_image):
     # Realizar inferência com o modelo
     outputs = model(**inputs)
     # Obter a classe prevista
-    predicted_class_id = np.argmax(outputs.logits[0])
+    predicted_class_id = np.argmax(outputs.logits[0].detach().numpy())
     # Obter o rótulo da classe a partir do mapeamento id2label
     predicted_class_label = id2label.get(str(predicted_class_id), "Desconhecido")
     
@@ -54,4 +54,4 @@ interface = gr.Interface(
 )
 
 # Iniciar a aplicação Gradio
-interface.launch()
+interface.launch(share=True)  # Compartilhar a interface com um link público