Update app.py

app.py

@@ -8,8 +8,7 @@ License: MIT License
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import gradio as gr
-import json  # Added for JSON conversion
-
+import json
 
 def run_inference(review_text: str) -> str:
     """
@@ -90,31 +89,24 @@ def predict_wine_variety(country: str, description: str, output_format: str) ->
 
 
 if __name__ == "__main__":
-    # Using Gradio Blocks to allow custom layout with a footer.
-    with gr.Blocks() as demo:
-        gr.Markdown(
-            """
-# Wine Variety Predictor
-
-Predict the wine variety based on the country and wine review.
-
-This tool uses ModernBERT, an encoder-only classifier, trained on the wine reviews dataset
-(model: spawn99/modernbert-wine-classification, dataset: spawn99/wine-reviews).
-
-Use the Output Format selector to toggle between a JSON-formatted result and a plain text prediction.
-            """
-        )
-        with gr.Row():
-            country_input = gr.Textbox(label="Country", placeholder="Enter country of origin...")
-            description_input = gr.Textbox(label="Description", placeholder="Enter wine review description...")
-            output_format_input = gr.Radio(choices=["JSON", "Text"], value="JSON", label="Output Format")
-        prediction_output = gr.Textbox(label="Prediction")
-        submit_btn = gr.Button("Predict")
-        submit_btn.click(
-            fn=predict_wine_variety,
-            inputs=[country_input, description_input, output_format_input],
-            outputs=prediction_output
+    iface = gr.Interface(
+        fn=predict_wine_variety,
+        inputs=[
+            gr.Textbox(label="Country", placeholder="Enter country of origin..."),
+            gr.Textbox(label="Description", placeholder="Enter wine review description..."),
+            gr.Radio(choices=["JSON", "Text"], value="JSON", label="Output Format")
+        ],
+        outputs=gr.Textbox(label="Prediction"),
+        title="Wine Variety Predictor",
+        description=(
+            "Predict the wine variety based on the country and wine review.\n\n"
+            "This tool uses [ModernBERT](https://huggingface.co/answerdotai/ModernBERT-base), "
+            "an encoder-only classifier, trained on the [wine reviews dataset]"
+            "(https://huggingface.co/datasets/spawn99/wine-reviews)\n\n"
+            "**Model Resources:**\n"
+            "- Model: [spawn99/modernbert-wine-classification](https://huggingface.co/spawn99/modernbert-wine-classification)\n"
+            "- Dataset: [spawn99/wine-reviews](https://huggingface.co/datasets/spawn99/wine-reviews)\n\n"
+            "*Cavit Erginsoy, 2025*\n"
         )
-        # Footer added at the bottom of the page
-        gr.Markdown("---\n**Cavit Erginsoy, 2025, MIT License**")
-    demo.launch()
+    )
+    iface.launch()