Update app.py

app.py

@@ -6,12 +6,12 @@ tokenizer = AutoTokenizer.from_pretrained("sambanovasystems/SambaLingo-Hungarian
 model = AutoModelForCausalLM.from_pretrained("sambanovasystems/SambaLingo-Hungarian-Chat", device_map="auto", torch_dtype="auto")
 
 # Create the pipeline with max_new_tokens parameter
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device_map="auto", use_fast=False, max_length=1024)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device_map="auto", use_fast=False, max_new_tokens=50)
 
 # Define the chat function
 def chat(question):
     messages = [{"role": "user", "content": question}]
-    prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+    prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True, truncation=True)
     outputs = pipe(prompt)[0]
     return outputs["generated_text"]