Change HF entry point and add examples

README.md

@@ -6,11 +6,13 @@ colorTo: green
 pinned: false
 sdk: gradio
 python_version: 3.11
-app_file: app.py
+app_file: app/gui.py
 datasets:
   - mrshu/amazonreviews
   - stanfordnlp/sentiment140
   - stanfordnlp/imdb
+models:
+  - spacy/en_core_web_sm
 ---
 
 # Sentiment Analysis

app.py

@@ -2,4 +2,4 @@
 
 from app.gui import launch_gui
 
-launch_gui("models/sentiment140_tfidf_ft-20000.pkl", share=False)
+launch_gui(share=False)

app/cli.py

@@ -29,9 +29,12 @@ def cli() -> None: ...
 )
 def gui(model_path: Path, share: bool) -> None:
     """Launch the Gradio GUI"""
+    import os
+
     from app.gui import launch_gui
 
-    launch_gui(model_path, share)
+    os.environ["MODEL_PATH"] = model_path.as_posix()
+    launch_gui(share)
 
 
 @cli.command()

app/gui.py

@@ -42,15 +42,21 @@ def sentiment_analysis(text: str) -> str:
 
 demo = gr.Interface(
     fn=sentiment_analysis,
-    inputs="text",
+    inputs=gr.Textbox(lines=10, label="Enter text here"),
     outputs="label",
     title="Sentiment Analysis",
+    description="Predict the sentiment of a given text.",
+    examples=[
+        ["I love the weather today!"],
+        ["You are a terrible person."],
+        ["The movie we watched was boring."],
+        ["This website is amazing!"],
+    ],
 )
 
 
-def launch_gui(model_path: str, share: bool) -> None:
+def launch_gui(share: bool) -> None:
     """Launch the Gradio GUI."""
-    os.environ["MODEL_PATH"] = model_path
     demo.launch(share=share)