Upload app.py

app.py

@@ -32,7 +32,7 @@ model.eval()
 tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/cosmo2-tokenizer")
 
 def generate_text(prompt, max_length=50):
-    input_ids = tokenizer.encode(prompt, return_tensors="pt").half()  # match model dtype
+    input_ids = tokenizer.encode(prompt, return_tensors="pt")
     with torch.no_grad():
         output_ids = model.generate(input_ids, max_length=max_length, do_sample=True)
     return tokenizer.decode(output_ids[0], skip_special_tokens=True)