ctn8176 commited on
Commit
c34dd87
·
verified ·
1 Parent(s): 6eb4256

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -55,8 +55,8 @@ def get_movie_info(movie_title):
55
  def generate_response(prompt):
56
  input_text_template = (
57
  "Hi! I am a gen AI bot powered by the Writer/palmyra-small model. "
58
- "I am here to give helpful, detailed, and polite answers to your movie inquiries. "
59
- f"USER: {prompt}"
60
  "Writer AI:"
61
  )
62
 
@@ -89,5 +89,11 @@ def chat_function(message, history):
89
  return response
90
 
91
  # Create Gradio Chat Interface
92
- chat_interface = gr.ChatInterface(chat_function)
 
 
 
 
 
 
93
  chat_interface.launch(share=True)
 
55
  def generate_response(prompt):
56
  input_text_template = (
57
  "Hi! I am a gen AI bot powered by the Writer/palmyra-small model. "
58
+ "I am here to give helpful, detailed, and polite answers to your movie inquiries.\n"
59
+ f"USER: {prompt}\n"
60
  "Writer AI:"
61
  )
62
 
 
89
  return response
90
 
91
  # Create Gradio Chat Interface
92
+ chat_interface = gr.ChatInterface(
93
+ chat_function,
94
+ textbox=gr.Textbox(placeholder="Type in a movie title", container=False, scale=7),
95
+ title="Palmyra-small, Movie Chatbot ",
96
+ description="Type in any movie title for more information",
97
+ theme="soft",
98
+ examples=["Oppenheimer", "Barbie", "Poor Things"],)
99
  chat_interface.launch(share=True)