Update app.py

app.py

@@ -15,14 +15,10 @@ def predict(image):
     with torch.no_grad():
         outputs = model(**inputs)
     
-    # Get the model's original outputs (e.g., logits or probabilities)
+    # Return the model's original outputs (e.g., logits or probabilities)
     predictions = outputs.logits
-
-    # Convert predictions to a list and round to 2 decimal places if necessary
-    predictions_list = predictions.tolist()
-    rounded_predictions = [[round(pred, 2) for pred in prediction] for prediction in predictions_list]
     
-    return rounded_predictions
+    return predictions
 
 # Create Gradio interface
 iface = gr.Interface(