Update app.py

app.py

@@ -20,7 +20,7 @@ def get_label(output):
     labels = ClassLabel(num_classes=7, names=['constant', 'linear', 'np', 'logn', 'quadratic', 'nlogn', 'cubic'])
     return labels.int2str(label)
     
-def complexity_estimation(gen_prompt, topk):
+def complexity_estimation(gen_prompt):
     pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
     output = pipe(gen_prompt)[0]
     # add label conversion to class
@@ -33,13 +33,6 @@ iface = gr.Interface(
     fn=complexity_estimation, 
     inputs=[
         gr.Textbox(lines=10, label="Input code"),
-        gr.inputs.Slider(
-            minimum=1,
-            maximum=3,
-            step=1,
-            default=1,
-            label="Number of results to return",
-        ),
     ],
     outputs=[
     gr.Textbox(label="Predicted complexity", lines=1) ,
@@ -47,7 +40,7 @@ iface = gr.Interface(
 ],
     examples=example,
     layout="vertical",
-    theme="peach",
+    theme="huggingface",
     description=description,
     title=title
 )