File size: 1,137 Bytes
5c540cd cbc25b3 23c5145 cbc25b3 5c540cd cbc25b3 |
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 |
from utils import my_key
import gradio
import google.generativeai as palm
import os
def gimi(user, history):
palm.configure(api_key = my_key)
defaults = {
'model': 'models/text-bison-001',
'temperature': 0.7,
'candidate_count': 1,
'top_k': 40,
'top_p': 0.95,
'max_output_tokens': 1024,
'stop_sequences': [],
'safety_settings': [
{"category": "HARM_CATEGORY_DEROGATORY", "threshold": 4},
{"category": "HARM_CATEGORY_TOXICITY", "threshold": 4},
{"category": "HARM_CATEGORY_VIOLENCE", "threshold": 4},
{"category": "HARM_CATEGORY_SEXUAL", "threshold": 4},
{"category": "HARM_CATEGORY_MEDICAL", "threshold": 4},
{"category": "HARM_CATEGORY_DANGEROUS", "threshold": 4},
],
}
prompt = f"""{user}"""
response = palm.generate_text(
**defaults,
prompt=prompt
)
gen_response=(response.result)
return gen_response
demo = gradio.ChatInterface(gimi)
demo.launch(share=True) |