Update app.py

app.py

@@ -51,7 +51,7 @@ model = CustomDialoGPT(vocab_size, n_embd, n_head, n_layer)
 # Download and load model weights
 try:
     pth_filepath = hf_hub_download(repo_id=model_repo, filename=model_filename)
-    checkpoint = torch.load(pth_filepath, map_location=device, weights_only=True)
+    checkpoint = torch.load(pth_filepath, map_location=device)
 
     # Handle different checkpoint saving formats if needed.
     # If your checkpoint is just the state_dict, load it directly.
@@ -113,10 +113,10 @@ iface = gr.Interface( # Changed from gr.ChatInterface to gr.Interface
     outputs=gr.Chatbot(), # Explicitly define outputs as gr.Chatbot
     title="ElapticAI-1a Chatbot",
     description="Simple chatbot interface for ElapticAI-1a model. Talk to the model and see its responses!",
-    examples=[
-        ["Hello"],
-        ["How are you?"],
-        ["Tell me a joke"]
+    examples=[ # Corrected examples format
+        ["Hello", "Hi there!"], # Example 1: [user_input, bot_response]
+        ["How are you?", "I am doing well, thank you."], # Example 2
+        ["Tell me a joke", "Why don't scientists trust atoms? Because they make up everything! 😄"] # Example 3
     ]
 )