NTThong0710 commited on
Commit
71209d1
·
1 Parent(s): 7ece9d3

Gemini fixing

Browse files
Files changed (2) hide show
  1. app/gen_ai.py +4 -7
  2. requirements.txt +1 -1
app/gen_ai.py CHANGED
@@ -6,13 +6,10 @@
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"]
 
6
  # result = generator(prompt, max_length=100, do_sample=True, temperature=0.7)
7
  # return result[0]["generated_text"]
8
 
9
+ from google import genai
10
 
11
+ client = genai.Client(api_key="AIzaSyAYEaSiZq7lcT5eJaVwnZJJ_UMoUTtpjJk")
 
 
 
12
 
13
  def generate_response(prompt: str):
14
+ result = client.models.generate_content(model="gemini-2.0-flash", contents="GEMINI")
15
+ return result.text
requirements.txt CHANGED
@@ -5,4 +5,4 @@ protobuf
5
  presidio-analyzer
6
  detoxify
7
  Pillow
8
- google-generativeai
 
5
  presidio-analyzer
6
  detoxify
7
  Pillow
8
+ google-genai