Update app.py

app.py

@@ -9,7 +9,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float
 
 def generate_text(input_text):
     # Ensure input is in the correct format
-    input_tensor = tokenizer(input_text, return_tensors="pt", clean_up_tokenization_spaces=True)
+    input_tensor = tokenizer(input_text, return_tensors="pt")  # Removed clean_up_tokenization_spaces
     
     # Generate text with a limit on max_length to reduce memory usage
     output = model.generate(**input_tensor, max_length=50)  # Adjust max_length as needed