Update app.py

app.py

@@ -68,6 +68,8 @@ def deepmojify(sentence, top_n=5, prob_only=False):
         tokenized = torch.tensor(tokenized).cuda()  # then convert to PyTorch tensor
 
     prob = emojimodel.forward(tokenized)[0]
+    if not USE_GPU:
+        prob = torch.tensor(prob)
     if prob_only:
         return prob
     emoji_ids = top_elements(prob.cpu().numpy(), top_n)