Update app.py

app.py

@@ -54,10 +54,10 @@ def get_movie_info(movie_title):
 
 def generate_response(prompt):
     input_text_template = (
-        "A chat between a curious user and an artificial intelligence assistant. "
-        "The assistant gives helpful, detailed, and polite answers to the user's questions. "
-        f"USER: {prompt} "
-        "ASSISTANT:"
+        "Hi! I am a gen AI bot powered by the Writer/palmyra-small model. "
+        "I am here to give helpful, detailed, and polite answers to your movie inquiries. "
+        f"USER: {prompt} (e.g., Oppenheimer, La La Land, Moonlight)"
+        "Writer AI:"
     )
 
     # Call the get_movie_info function to enrich the response
@@ -80,7 +80,7 @@ def generate_response(prompt):
 
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
     
-    return f"Movie Info:\n{movie_info}\n\nGenerated Response:\n{generated_text}\n"
+    return f"Movie Info:\n{movie_info}\n\n Writer AI Generated Response:\n{generated_text}\n"
 
 # Define chat function for gr.ChatInterface
 def chat_function(message, history):
@@ -90,4 +90,4 @@ def chat_function(message, history):
 
 # Create Gradio Chat Interface
 chat_interface = gr.ChatInterface(chat_function)
-chat_interface.launch(share=True)  
+chat_interface.launch(share=True, label='Try something like: Oppenheimer, La La Land, Moonlight')