Update app.py

app.py

@@ -1,3 +1,10 @@
+"""
+Wine Variety Predictor
+Author: Cavit Erginsoy
+Year: 2025
+License: MIT License
+"""
+
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import gradio as gr
@@ -83,22 +90,31 @@ def predict_wine_variety(country: str, description: str, output_format: str) ->
 
 
 if __name__ == "__main__":
-    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..."),
-            # New radio input to choose between JSON and plain text output formats:
-            gr.Radio(choices=["JSON", "Text"], value="JSON", label="Output Format")
-        ],
-        # Changed outputs to a Textbox so that plain text output shows naturally
-        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, an encoder-only classifier, trained on the wine reviews dataset\n"
-            "(model: spawn99/modernbert-wine-classification, dataset: spawn99/wine-reviews).\n\n"
-            "Use the Output Format selector to toggle between a JSON-formatted result and a plain text prediction."
+    # 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.
+            """
         )
-    )
-    iface.launch()
+        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
+        )
+        # Footer added at the bottom of the page
+        gr.Markdown("---\n**Cavit Erginsoy, 2025, MIT License**")
+    demo.launch()