Update app.py

app.py

@@ -16,4 +16,4 @@ def predict_fake(title,text):
         output = model(input_ids["input_ids"].to(device), attention_mask=input_ids["attention_mask"].to(device))
     return dict(zip(["Fake","Real"], [x.item() for x in list(torch.nn.Softmax()(output.logits)[0])] ))
 
-iface = gr.Interface(fn=predict_fake, inputs=[gr.inputs.Textbox(lines=1,label="headline"),gr.inputs.Textbox(lines=6,label="content")], outputs="label").launch(share=True)
+gr.Interface(fn=predict_fake, inputs=[gr.Textbox(lines=1,label="headline"), gr.Textbox(lines=6,label="content")], outputs=gr.Label()).launch(share=True)