Update app.py

app.py

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