File size: 583 Bytes
8e479b1
487c749
8e479b1
487c749
6121e18
01cf111
487c749
 
 
 
 
 
 
 
 
 
 
 
 
01cf111
487c749
01cf111
8e479b1
01cf111
8e479b1
 
 
 
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
import gradio as gr
import google.generativeai as genai
from huggingface_hub import InferenceClient
import gradio as gr

client = InferenceClient("Google/gemini-1.5-flash")
genai.configure(api_key=os.environ["GEMINI_API_KEY"])

generation_config = {
  "temperature": 0.7,
  "top_p": 0.95,
  "top_k": 64,
  "max_output_tokens": 128,
  "response_mime_type": "text/plain",
}

model = genai.GenerativeModel(
  model_name="gemini-1.5-flash",
  generation_config=generation_config,
  
)
response = model.generate_content

print(response.text)


if __name__ == "__main__":
    demo.launch()