Update app.py

app.py

@@ -12,5 +12,13 @@ def generate(commentary_text):
     output = trained_model.generate(input_ids, max_length=60, num_beams=5, do_sample=False)
     return tokenizer_finetuned.decode(output[0])
 
-demo = gr.Interface(fn=generate, inputs="text", outputs="text")
-demo.launch() 
+# Create Gradio interface
+iface = gr.Interface(fn=generate_text, 
+                     inputs="text", 
+                     outputs="text",
+                     title="GPT-2 Text Generation",
+                     description="Enter a prompt and GPT-2 will generate the continuation of the text.")
+
+# Launch the app
+if __name__ == "__main__":
+    iface.launch()