Update chatbot.py

chatbot.py

@@ -46,7 +46,7 @@ def load_pipeline():
     pipe = pipeline("text-generation", 
                     model= model, 
                     tokenizer = tokenizer,
-                    max_new_tokens = 20,
+                    max_new_tokens = 50,
                     top_k = 30, 
                     early_stopping=True,
                     num_beams = 2,
@@ -78,4 +78,4 @@ def demo_chain(input_text,history):
     )
 
     chat_reply = conversation.invoke(input=input_text, max_new_tokens = 100)
-    return chat_reply
+    return chat_reply['response'].split('AI:')[-1]