Update app.py

app.py

@@ -1,4 +1,4 @@
-import torch
+import torch 
 from transformers import *
 
 # Choose the device and load the model
@@ -19,5 +19,7 @@ input_ids = encoded_text['input_ids'].to(device)
 attention_mask = encoded_text['attention_mask'].to(device) 
 
 
-outputs = model(input_ids, attention_mask=attention_mask) 
-predictions = outputs[0]
+
+ outputs = model(input_ids, attention_mask=attention_mask) 
+
+ predictions = outputs[0]