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 os
 import torch
 
 app = Flask(__name__)
@@ -18,8 +19,8 @@ def upload_image():
     image_file = request.files['image']
 
     # Save the image file to a desired location on the server
-    image_path = "assets/img.jpg"
-    image_file.save(image_path)
+    # image_path = "assets/"
+    image_file.save(os.path.join('/', 'img.jpg'))
 
     # You can perform additional operations with the image here
     # ...
@@ -29,7 +30,7 @@ def upload_image():
 
 @app.route('/get_text', methods=['GET'])
 def get_text():
-    image = Image.open('assets/img.jpg')
+    image = Image.open('/img.jpg')
     inputs = image_processor(image, return_tensors="pt")
 
     with torch.no_grad():