File size: 1,005 Bytes
d36cdf1 1f42217 001d7ee d36cdf1 937cd3c 77d2a81 d36cdf1 b7366a2 937cd3c 4c3d295 d36cdf1 1f42217 d36cdf1 001d7ee 13c61e3 001d7ee d36cdf1 13c61e3 d36cdf1 13c61e3 21e00ad 937cd3c d36cdf1 937cd3c d36cdf1 |
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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
from flask import Flask, render_template, request, jsonify
import requests
import os
from flask_cors import CORS
app = Flask(__name__)
CORS(app)
HF_MODEL = "google/gemma-2b"
HF_API_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
# Lấy API Key từ Hugging Face Secrets
HF_API_KEY = os.getenv("HF_API_KEY")
# Gửi request API để chat
def chat_with_gemma(prompt):
if not HF_API_KEY:
return "Lỗi: API Key không tồn tại. Hãy thêm vào Hugging Face Secrets."
response = requests.post(
HF_API_URL,
headers={"Authorization": f"Bearer {HF_API_KEY}"},
json={"inputs": prompt}
)
if response.status_code == 200:
return response.json()[0].get('generated_text', "Không có phản hồi.")
return f"Lỗi: {response.status_code} - {response.text}"
@app.route('/')
def index():
return jsonify({'FROM': 'Gemma-2B', 'MESSAGE': bot_reply})
if __name__ == '__main__':
app.run(host="0.0.0.0", port=7860)
|