removed cache

app.py

@@ -9,25 +9,16 @@ from random import randint
 def predict_cyberbullying_probability(sentence, tokenizer, model):
     # Preprocess the input sentence
     inputs = tokenizer(sentence, padding='max_length', return_token_type_ids=False, return_attention_mask=True, truncation=True, max_length=512, return_tensors='pt')
-
-    print("==========")
-    print(inputs)
-    print("==========")
     
-    attention_mask = inputs['attention_mask'].flatten()
-    print("==========")
-    print(attention_mask)
-    inputs = inputs['input_ids'].flatten()
-    # print("\n\ninputs\n\n", inputs)
-    # Disable gradient computation
+    attention_mask = inputs['attention_mask']
+    inputs = inputs['input_ids']
+    
     with torch.no_grad():
         # Forward pass
         outputs = model(inputs, attention_mask=attention_mask)
 
         probs = torch.sigmoid(outputs.logits.unsqueeze(1).flatten())
 
-        
-    
     res = probs.numpy().tolist()
     return res