Update app.py

app.py

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