Spaces:
Sleeping
Sleeping
import os | |
from dotenv import load_dotenv | |
from flask import Flask, request, jsonify | |
import requests | |
load_dotenv() | |
api_key = os.getenv('HF_API_KEY') | |
model_path = os.getenv('MODEL_PATH') | |
app = Flask(__name__) | |
def get_model_predictions(text): | |
headers = {"Authorization": f"Bearer {api_key}"} | |
payload = {"inputs": text} | |
response = requests.post(f"https://api.huggingface.co/models/{model_path}", headers=headers, json=payload) | |
return response.json() | |
def predict(): | |
data = request.get_json() | |
text = data['text'] | |
prediction = get_model_predictions(text) | |
return jsonify(prediction) | |
def index(): | |
return "Welcome to My AI!" | |
if __name__ == '__main__': | |
app.run(debug=True) |