Update app.py

app.py

@@ -107,10 +107,23 @@ iface = gr.TabbedInterface([iface_sentence, iface_dataset], ["Sentence", "Datase
 iface.queue().launch()
 """
 
+inference_modelpath = "model/checkpoint-128"
 
+def inference_sentence(text):
+    tokenizer = AutoTokenizer.from_pretrained(inference_modelpath)
+    model = AutoModelForSequenceClassification.from_pretrained(inference_modelpath)
+    for text in tqdm([text]):
+        inputs = tokenizer(text, return_tensors="pt")
+    with torch.no_grad(): # run model
+        logits = model(**inputs).logits
+        predicted_class_id = logits.argmax().item()
+    output = model.config.id2label[predicted_class_id]
+    return output
+"""
 def inference_sentence(text):
     output = "This sentence will be processed:\n" + text
     return output
+"""
 
 def unavailable(input_file, input_checks):
     output = "Submitting your own data is currently unavailable, but you can try out the showcase mode 😊"