files changed

app.py

@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5", trust_remote_code
 
 def infer(text):
     inputs = tokenizer(f'''question: {text} answer: ''', return_tensors="pt", return_attention_mask=False)
-    outputs = model.generate(**inputs, max_length=116)
+    outputs = model.generate(**inputs, max_length=156)
     text = tokenizer.batch_decode(outputs)[0]
     text = text.split('answer:',1)[1]
     return text