| 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) |