Update app.py

app.py

@@ -81,8 +81,12 @@ def update_output(model_name: str, text: str, custom_token: str, selected_token:
     
     return output, "", "", gr.update(choices=token_choices), predictions
 
-with gr.Blocks() as app:
+# Create the interface
+demo = gr.Blocks(title="Interactive Text Generation")
+
+with demo:
     gr.Markdown("# Interactive Text Generation")
+    gr.Markdown("Generate text by selecting predicted tokens or writing your own.")
     
     with gr.Row():
         model_dropdown = gr.Dropdown(
@@ -128,5 +132,5 @@ with gr.Blocks() as app:
             outputs=[text_input, custom_token, token_dropdown, token_dropdown, predictions_output]
         )
 
-if __name__ == "__main__":
-    app.launch(share=True)
+# For Hugging Face Spaces, we just need to expose the demo
+demo.launch()