Spaces:
Sleeping
Sleeping
NTThong0710 commited on
Commit ·
7ece9d3
1
Parent(s): 8bab17a
Try Gemini
Browse files- .github/workflows/deploy.yml +30 -0
- app/gen_ai.py +15 -4
- requirements.txt +1 -0
.github/workflows/deploy.yml
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: CI/CD for Python App
|
| 2 |
+
|
| 3 |
+
on:
|
| 4 |
+
push:
|
| 5 |
+
branches: [main] # hoặc 'master'
|
| 6 |
+
|
| 7 |
+
jobs:
|
| 8 |
+
deploy:
|
| 9 |
+
runs-on: ubuntu-latest
|
| 10 |
+
|
| 11 |
+
steps:
|
| 12 |
+
- name: 📥 Lấy mã nguồn
|
| 13 |
+
uses: actions/checkout@v3
|
| 14 |
+
|
| 15 |
+
- name: 🐍 Cài Python
|
| 16 |
+
uses: actions/setup-python@v4
|
| 17 |
+
with:
|
| 18 |
+
python-version: "3.11"
|
| 19 |
+
|
| 20 |
+
- name: 📦 Cài dependencies
|
| 21 |
+
run: |
|
| 22 |
+
python -m pip install --upgrade pip
|
| 23 |
+
pip install -r requirements.txt
|
| 24 |
+
|
| 25 |
+
- name: ✅ Kiểm tra (test)
|
| 26 |
+
run: |
|
| 27 |
+
pytest || echo "⚠️ Bỏ qua test nếu chưa có"
|
| 28 |
+
|
| 29 |
+
- name: 🚀 Deploy lên Huggingface
|
| 30 |
+
run: curl -X POST ${{ secrets.DEPLOY_HOOK_URL }}
|
app/gen_ai.py
CHANGED
|
@@ -1,7 +1,18 @@
|
|
| 1 |
-
from transformers import pipeline
|
| 2 |
|
| 3 |
-
generator = pipeline("text-generation", model="gpt2")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
def generate_response(prompt: str):
|
| 6 |
-
result =
|
| 7 |
-
return result[0]["generated_text"]
|
|
|
|
| 1 |
+
# from transformers import pipeline
|
| 2 |
|
| 3 |
+
# generator = pipeline("text-generation", model="gpt2")
|
| 4 |
+
|
| 5 |
+
# def generate_response(prompt: str):
|
| 6 |
+
# result = generator(prompt, max_length=100, do_sample=True, temperature=0.7)
|
| 7 |
+
# return result[0]["generated_text"]
|
| 8 |
+
|
| 9 |
+
import google.generativeai as genai
|
| 10 |
+
|
| 11 |
+
API_KEY = "AIzaSyAYEaSiZq7lcT5eJaVwnZJJ_UMoUTtpjJk"
|
| 12 |
+
genai.configure(api_key=API_KEY)
|
| 13 |
+
|
| 14 |
+
model = genai.GenerativeMode("gemini-2.0-flash")
|
| 15 |
|
| 16 |
def generate_response(prompt: str):
|
| 17 |
+
result = model(prompt, max_length=100, do_sample=True, temperature=0.7)
|
| 18 |
+
return result[0]["generated_text"]
|
requirements.txt
CHANGED
|
@@ -5,3 +5,4 @@ protobuf
|
|
| 5 |
presidio-analyzer
|
| 6 |
detoxify
|
| 7 |
Pillow
|
|
|
|
|
|
| 5 |
presidio-analyzer
|
| 6 |
detoxify
|
| 7 |
Pillow
|
| 8 |
+
google-generativeai
|