Update app.py

app.py

@@ -106,7 +106,7 @@ def main():
 
         if pred_button:
             
-            outputs = float(model.predict(image))
+            outputs = int(model.predict(image))
             _, y_hat = outputs.max(1)
             predicted_idx = str(y_hat.item())
             right_column.title("Prediction")