Update app.py

app.py

@@ -29,12 +29,13 @@ model = AutoModelForCausalLM.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 
 # Example usage
-input_text = "Once upon a time"
-input_ids = tokenizer(input_text, return_tensors="pt").input_ids
-output = model.generate(input_ids, max_length=50, num_return_sequences=1)
-generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
-
-print(generated_text)
+# input_text = "Once upon a time"
+def generated_text(input_text):
+    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
+    output = model.generate(input_ids, max_length=50, num_return_sequences=1)
+    generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+    return generated_text
+# print(generated_text)
 
 
 text_generation_interface = gr.Interface(