| # -*- coding: utf-8 -*- | |
| """Chatbot Using Palm2.ipynb | |
| Automatically generated by Colaboratory. | |
| Original file is located at | |
| https://colab.research.google.com/drive/1LV6u866sFrGHdL6i_zQlTmtnikCJ15il | |
| """ | |
| import google.generativeai as palm | |
| import gradio as gr | |
| palm.configure(api_key='AIzaSyDa89CpRmIwKA6h8fBO533Si0xK_YvSs7I') | |
| models = [m for m in palm.list_models() if 'generateText' in m.supported_generation_methods] | |
| model = models[0].name | |
| def get_text_response(user_message,history): | |
| completion = palm.generate_text( | |
| model=model, | |
| prompt=user_message, | |
| temperature=0.25, # temperature=0 >> more deterministic results // temperature=1 >> more randomness | |
| max_output_tokens=100 # maximum length of response | |
| ) | |
| return completion.result | |
| demo = gr.ChatInterface(get_text_response, examples=["How are you doing?","What are your interests?","Which places do you like to visit?"]) | |
| if __name__ == "__main__": | |
| demo.launch() |