Update app.py

app.py

@@ -26,7 +26,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)
+    predicted_class_id = np.argmax(outputs.logits[0])
     # Obter o rótulo da classe a partir do mapeamento id2label
     predicted_class_label = id2label.get(str(predicted_class_id), "Desconhecido")
     return predicted_class_label