Update app.py

app.py

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