Update app.py

app.py

@@ -14,9 +14,9 @@ model_name = "allenai/OLMo-1B"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 
-def generate_text(prompt, max_new_tokens, top_k, top_p, do_sample):
+def generate_text(prompt, max_new_tokens=400, do_sample=True, top_k=50, top_p=0.95):
     inputs = tokenizer(prompt, return_tensors='pt', return_token_type_ids=False)
-    response = model.generate(**inputs, max_new_tokens=400, do_sample=Truee, top_k=50, top_p=0.95)
+    response = model.generate(**inputs, max_new_tokens, do_sample, top_k, top_p)
     return tokenizer.batch_decode(response, skip_special_tokens=True)[0]