add slider

app.py

@@ -7,14 +7,14 @@ tokenizer = AutoTokenizer.from_pretrained("distilbert/distilbert-base-uncased")
 model = AutoModelForSequenceClassification.from_pretrained("usman-yello/job-tagging")
 
 
-def tag_job_description(job_description):
+def tag_job_description(job_description, num_tags):
     job_description = re.sub(r'[^a-zA-Z]', ' ', job_description)
     inputs = tokenizer(job_description, truncation=True, max_length=512, return_tensors="pt")
     outputs = model(**inputs)
-    topk_values, topk_indices = torch.topk(outputs.logits, 3)
-    predicted_tags = [model.config.id2label[i] for i in topk_indices[0]]
+    topk_values, topk_indices = torch.topk(outputs.logits, num_tags)
+    predicted_tags = [model.config.id2label[i.item()] for i in topk_indices[0]]
     return predicted_tags
 
 
-demo = gr.Interface(fn=tag_job_description, inputs="text", outputs="text")
-demo.launch()
+job_tagger = gr.Interface(fn=tag_job_description, inputs=["text", gr.Slider(value=1, minimum=1, maximum=10, step=1)], outputs="text")
+job_tagger.launch()