Update app.py

app.py

@@ -78,12 +78,12 @@ def image_transformation(image):
     return image
 
 
-def image_prediction(image, model):
-    image = image_transformation(image=image)
-    outputs = double(model.predict(image))
-    _, y_hat = outputs.max(1)
-    predicted_idx = str(y_hat.item())
-    return predicted_idx
+# def image_prediction(image, model):
+#     image = image_transformation(image=image)
+#     outputs = float(model.predict(image))
+#     _, y_hat = outputs.max(1)
+#     predicted_idx = str(y_hat.item())
+#     return predicted_idx
 
 def main():
     
@@ -105,8 +105,8 @@ def main():
         # labels = ohe.fit_transform(label).toarray()
 
         if pred_button:
-            image_prediction(image=image, model=model)
-            outputs = double(model.predict(image_prediction))
+            
+            outputs = float(model.predict(image))
             _, y_hat = outputs.max(1)
             predicted_idx = str(y_hat.item())
             right_column.title("Prediction")