test

Dockerfile

@@ -8,6 +8,9 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
 
+# Move the Hugging Face cache directory
+RUN mv ~/.cache/huggingface/transformers ~/.cache/huggingface-tmp || true
+
 COPY . .
 
 CMD ["gunicorn", "-b", "0.0.0.0:7860", "main:app"]

main.py

@@ -1,11 +1,17 @@
 from flask import Flask, jsonify
+from flask_cors import CORS
 from transformers import pipeline
 
 app = Flask(__name__)
+CORS(app)
 
 # Load the sentiment-analysis pipeline from Hugging Face
 sentiment_analyzer = pipeline('sentiment-analysis')
 
+@app.route('/')
+def hello():
+    return {"Goes Wrong": "Keeping it real"}
+
 @app.route('/analyze', methods=['GET'])
 def analyze():
     sentence = "hey there"

requirements.txt

@@ -13,4 +13,5 @@ matplotlib  # Required for image processing and mask visualization
 onnxruntime
 onnx
 pycocotools
-transformers
+transformers
+torch