Update app.py

app.py

@@ -13,7 +13,7 @@ pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
 def generate_response(prompt):
     # Format the prompt as required by the model
     input_text = f"<s>[INST] {prompt} [/INST]"
-    response = pipe(input_text, max_length=200)
+    response = pipe(input_text, max_new_tokens=200)
     # Extract the generated text from the response
     answer = response[0]['generated_text'].split('[/INST]')[-1].strip()
     return answer