Update app.py

app.py

@@ -24,13 +24,13 @@ def generate_full_story(excerpt: str) -> str:
     # Generate tokens. Here, we set parameters to control length and creativity.
     output = model.generate(
         encoded_input["input_ids"],
-        attention_mask=encoded_input["attention_mask"],
-        max_new_tokens=200,         # Generate 200 new tokens on top of the input.
-        do_sample=True,
-        temperature=0.8,
-        top_p=0.95,
-        no_repeat_ngram_size=2,
-        return_dict_in_generate=True
+        attention_mask = encoded_input["attention_mask"],
+        max_new_tokens = 200,         # Generate 200 new tokens on top of the input.
+        do_sample = True,
+        temperature = 0.8,
+        top_p = 0.95,
+        no_repeat_ngram_size = 2,
+        return_dict_in_generate = True
     )
     
     # Decode the generated sequence.
@@ -40,20 +40,19 @@ def generate_full_story(excerpt: str) -> str:
 
 # Build the Gradio interface.
 interface = gr.Interface(
-    fn=generate_full_story,
-    inputs=gr.Textbox(
-        lines=5,
-        label="Incomplete Story Excerpt",
-        placeholder="Enter your incomplete story excerpt here..."
+    fn = generate_full_story,
+    inputs = gr.Textbox(
+        lines = 5,
+        label = "Incomplete Story Excerpt",
+        placeholder = "Enter your incomplete story excerpt here..."
     ),
-    outputs=gr.Textbox(label="Complete Story with Details"),
+    outputs=gr.Textbox(label = "Complete Story with Details"),
     title="Mythology Storyteller",
     description=(
         "Enter an incomplete story excerpt. "
-        "The model will generate a complete output that includes the chapter (Parv), key event, section, "
-        "and the full story continuation."
+        "The model will generate a complete the complete chapter."
     )
 )
 
 # Launch the Gradio app.
-interface.launch(share=True)
+interface.launch(share = True)