test

main.py

@@ -1,38 +1,16 @@
-import os
-from flask import Flask, jsonify, request
-from flask_cors import CORS
+from flask import Flask, jsonify
 from transformers import pipeline
 
-# Set the TRANSFORMERS_CACHE environment variable to a writable directory
-os.environ['TRANSFORMERS_CACHE'] = '/tmp/huggingface/cache'
-
 app = Flask(__name__)
-CORS(app)
-
-# Initialize the sentiment analysis pipeline
-classifier = pipeline("sentiment-analysis")
 
-@app.route('/')
-def hello():
-    return 'hello world'
+# Load the sentiment-analysis pipeline from Hugging Face
+sentiment_analyzer = pipeline('sentiment-analysis')
 
-@app.route('/analyze-sentiment', methods=['POST'])
-def analyze_sentiment():
-    try:
-        # Get the text from the request
-        data = request.get_json()
-        if 'text' not in data:
-            return jsonify({"error": "No text provided"}), 400
-        
-        text = data['text']
-        # Run sentiment analysis
-        result = classifier(text)
-        return jsonify(result)
-    except Exception as e:
-        # Log the error message if needed
-        print(f"Error processing the text: {e}")
-        # Return a JSON response with the error message and a 400 Bad Request status
-        return jsonify({"error": "Error processing the text", "details": str(e)}), 400
+@app.route('/analyze', methods=['GET'])
+def analyze():
+    sentence = "hey there"
+    result = sentiment_analyzer(sentence)
+    return jsonify(result)
 
 if __name__ == '__main__':
     app.run(debug=True)

requirements.txt

@@ -1,16 +1,16 @@
 flask
 gunicorn
 flask-cors
-transformers
-# numpy
-# opencv-python
-# Pillow
-# requests
-# git+https://github.com/facebookresearch/segment-anything.git
-# --extra-index-url https://download.pytorch.org/whl/cu113
-# torch
-# torchvision
-# matplotlib  # Required for image processing and mask visualization
-# onnxruntime
-# onnx
-# pycocotools
+numpy
+opencv-python
+Pillow
+requests
+git+https://github.com/facebookresearch/segment-anything.git
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
+torchvision
+matplotlib  # Required for image processing and mask visualization
+onnxruntime
+onnx
+pycocotools
+transformers