NTThong0710 commited on
Commit
7ece9d3
·
1 Parent(s): 8bab17a

Try Gemini

Browse files
Files changed (3) hide show
  1. .github/workflows/deploy.yml +30 -0
  2. app/gen_ai.py +15 -4
  3. 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 = generator(prompt, max_length=100, do_sample=True, temperature=0.7)
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