Update app.py

app.py

@@ -2,6 +2,7 @@ from flask import Flask, request
 from transformers import AutoModelForImageClassification
 from transformers import AutoImageProcessor
 from PIL import Image
+import BytesIO
 import os
 import torch
 
@@ -20,27 +21,32 @@ def upload_image():
 
     # Save the image file to a desired location on the server
     # image_path = "assets/"
-    image_file.save(os.path.join('/', 'img.jpg'))
-
-    # You can perform additional operations with the image here
-    # ...
-
-    return "Image uploaded successfully"
-
-
-@app.route('/get_text', methods=['GET'])
-def get_text():
-    image = Image.open('/img.jpg')
+    # image_file.save(os.path.join('/', 'img.jpg'))
+    
+    image = Image.open(BytesIO(image_file))
     inputs = image_processor(image, return_tensors="pt")
-
+    
     with torch.no_grad():
         logits = model(**inputs).logits
 
     predicted_label = logits.argmax(-1).item()
 
     disease = model.config.id2label[predicted_label]
+    
 
+    # You can perform additional operations with the image here
+    # ...
+
+    # return "Image uploaded successfully"
     return disease
+    
+
+
+@app.route('/get_text', methods=['GET'])
+def get_text():
+    # image = Image.open('/img.jpg')
+    return
+
 
 
 @app.route('/', methods=['GET'])