Spaces:
Running
Running
File size: 772 Bytes
a471f05 7e8513d a471f05 7106645 7e8513d 359f489 7e8513d 7106645 7e8513d 7106645 00c6e88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import os
# Default /.cache/huggingface ci vengono scaricati i modelli
os.environ["HF_HOME"] = "/tmp/huggingface"
from flask import Flask, jsonify, request
from flask_cors import CORS
from model.sentiment import Sentiment
from urllib.parse import unquote
app = Flask(__name__)
CORS(app)
# Endpoint '/'
@app.route('/')
def home():
return jsonify({"message": "API is running!"})
# Endpoint '/analisis'
@app.route('/analisis', methods=['GET'])
def analisis():
prompt = request.args.get('prompt', '')
prompt = unquote(prompt)
if not prompt:
return jsonify({"error": "No prompt provided"}), 400
model = Sentiment(prompt)
return jsonify({"message": model.result}), 200
if __name__ == '__main__':
app.run(host='0.0.0.0', port=7860) |