|
|
|
"""Copy of Chatbot Using Palm2.ipynb |
|
|
|
Automatically generated by Colaboratory. |
|
|
|
Original file is located at |
|
https://colab.research.google.com/drive/1LmFbsKVPTLlpcFU1oiC0wTfgE4mxeMs9 |
|
""" |
|
|
|
|
|
import google.generativeai as palm |
|
import gradio as gr |
|
|
|
palm.configure(api_key='AIzaSyAtWxs2QIiiWfvKaIUxoVEfUOF5fnLWA60') |
|
|
|
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, |
|
|
|
max_output_tokens=100 |
|
|
|
) |
|
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() |