Update: Remove mode option

app.py

@@ -113,7 +113,7 @@ def is_eng(review: str):
 
 ### Do actual prediction #########################################################
 @spaces.GPU(duration=120)  # Sekunden GPU-Zeit pro Call
-def predict(review: str, mode: str):
+def predict(review: str):
 
     review = (review or "").strip()
 
@@ -131,35 +131,30 @@ def predict(review: str, mode: str):
         prediction_flavours = predict_flavours(review, model_flavours, tokenizer_flavours, device)
         t_end_flavours = time.time()
 
-    if mode == "table":
-        html_out = f"<b>{html.escape(review)}</b> | <i>{mode}</i>"
-        json_out = {}  # leer lassen
-        return html_out, json_out
-    else:  # "JSON"
-        html_out = ""  # leer lassen
-        json_out = {
-            "review": review,
-            "mode": mode,
-            'model': FILENAME,
-            "device": device,
-            "duration": round((t_end_flavours - t_start_flavours), 3),
-            "is_en": {
-                "is": review_is_eng,
-                "prob": review_is_eng_prob
-            },
-            "results": [{
-                    'icon': ICONS[name],
-                    'name': name,
-                    'score': score,
-                    #'level': get_level(score)
-                }
-                for name, score in prediction_flavours.items()
-            ],
-        }
-        return html_out, json_out
+    html_out = f"<b>{html.escape(review)}</b>"
+
+    json_out = {
+        "review": review,
+        'model': FILENAME,
+        "device": device,
+        "duration": round((t_end_flavours - t_start_flavours), 3),
+        "is_en": {
+            "is": review_is_eng,
+            "prob": review_is_eng_prob
+        },
+        "results": [{
+                'icon': ICONS[name],
+                'name': name,
+                'score': score,
+                #'level': get_level(score)
+            }
+            for name, score in prediction_flavours.items()
+        ],
+    }
+
+    return html_out, json_out
     
 ### Create Form interface with Gradio Framework ##################################
-
 iface = gr.Interface(
     fn=predict,
     inputs=[
@@ -168,19 +163,13 @@ iface = gr.Interface(
             lines=8, 
             placeholder="Enter whisky review",
             value="Honey roasted figs, cloves and sticky toffee pudding with a suggestive smokiness evocative of barbecued bananas."
-        ),
-        gr.Dropdown(
-            label="Display mode",
-            choices=[("Table", "table"), ("JSON", "json")],
-            value="table"
         )
     ],
     outputs=[
         gr.HTML(label="Table"),
         gr.JSON(label="JSON"),
     ],
-    title="Submit Whisky Review for Classification",
-    # description="Paste an English whisky review, choose display mode, then submit.",
+    title="Submit Whisky Review for Classification"
 )
 
 iface.launch()