Update app.py

app.py

@@ -1,6 +1,29 @@
 import gradio as gr
-import os
+from os import environ
+from transformers import pipeline
 
-api_key = os.environ.get("api_key")
+api_key = environ.get("api_key")
 
-gr.Interface.load("models/gianma/testModel", api_key=api_key).launch()
+def app(input, only_relevant_values, relevence_threshold, k ):
+    classifier_pipeline = pipeline(model="gianma/testModel", tokenizer='gianma/testModel', api_key=api_key)
+    
+    kwargs = {'padding':True,'truncation':True,'max_length':512}
+
+    if not only_relevant_values:
+        kwargs['top_k'] = k
+    
+    res = classifier(input, **kwargs)
+
+    if only_relevant_values:
+        res = {k:v for k,v in res.items() if k >= relevence_threshold}
+
+    return res
+
+iface = gr.Interface(fn=app, 
+                     inputs=["text", "checkbox", gr.Slider(0, 1), gr.Slider(1, 21)], 
+                     outputs="label")
+
+iface.launch()
+
+
+#gr.Interface.load("models/gianma/testModel", api_key=api_key).launch()