Fixed typo

app.py

@@ -20,13 +20,14 @@ def pred_is_elon_musk(text):
     masks = torch.Tensor([masks]).to(device, dtype=torch.int32)
     results = model(input_ids=ids, token_type_ids=None,
                     attention_mask=masks)
-    logis = results['logits']
+    logis = results['logits'].detach()
     prob = F.softmax(logis, dim=1)[0]
-    prediction = np.argmax(logis.detach().numpy(), axis=1).flatten()
+    prediction = np.argmax(logis.numpy(), axis=1).flatten()
     if prediction[0] == 1:
-        return f"It's from Elon Musk with {prob[0]*100: 0.2f}% probability."
+        output = f"It's from Elon Musk with {prob[0]*100: 0.2f}% probability."
     else:
-        return "It's NOT from Elon Musk with {prob[1]*100: 0.2f}% probability."
+        output = f"It's NOT from Elon Musk with {prob[1]*100: 0.2f}% probability."
+    return output
 
 iface = gr.Interface(pred_is_elon_musk, inputs="text", 
     outputs="text", title='“Is the tweet from Elon Musk?” Classifier',