Upload 2 files

app.py

@@ -0,0 +1,46 @@
+from flask import Flask, request
+from transformers import AutoModelForImageClassification
+from transformers import AutoImageProcessor
+from PIL import Image
+import torch
+
+app = Flask(__name__)
+
+model = AutoModelForImageClassification.from_pretrained(
+    '../apiModel/myModel')
+image_processor = AutoImageProcessor.from_pretrained(
+    "google/vit-base-patch16-224-in21k")
+
+
+@app.route('/upload_image', methods=['POST'])
+def upload_image():
+    # Get the image file from the request
+    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)
+
+    # 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('assets/img.jpg')
+    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]
+
+    return disease
+
+
+if __name__ == '__app__':
+    app.run(host='192.168.1.7', port=5000)

requirements.txt

@@ -0,0 +1,48 @@
+autopep8==1.7.0
+biopython==1.79
+blinker==1.6.2
+certifi==2023.5.7
+charset-normalizer==3.1.0
+click==8.1.3
+colorama==0.4.6
+contourpy==1.0.6
+cycler==0.11.0
+docopt==0.6.2
+filelock==3.12.0
+Flask==2.3.2
+fonttools==4.38.0
+fsspec==2023.5.0
+huggingface-hub==0.14.1
+idna==3.4
+itsdangerous==2.1.2
+Jinja2==3.1.2
+kiwisolver==1.4.4
+MarkupSafe==2.1.2
+matplotlib==3.6.1
+mpmath==1.3.0
+networkx==3.1
+numpy==1.23.4
+opencv-python==4.6.0.66
+packaging==21.3
+pandas==1.5.3
+Pillow==9.3.0
+pipreqs==0.4.13
+pycodestyle==2.9.1
+pyparsing==3.0.9
+python-dateutil==2.8.2
+pytz==2022.7.1
+PyYAML==6.0
+regex==2023.5.5
+requests==2.31.0
+scipy==1.9.3
+six==1.16.0
+sympy==1.12
+tokenizers==0.13.3
+toml==0.10.2
+torch==2.0.1
+tqdm==4.65.0
+transformers==4.29.2
+typing_extensions==4.5.0
+urllib3==2.0.2
+Werkzeug==2.3.4
+yarg==0.1.9