Update app.py

app.py

@@ -11,8 +11,8 @@ headers = {"Authorization": f"Bearer {hf_token}"}
 
 
 model_name = "allenai/OLMo-1B"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+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):
     inputs = tokenizer(prompt, return_tensors='pt', return_token_type_ids=False)