Update app.py

app.py

@@ -2,16 +2,13 @@
 #python -m streamlit run d:/NSFW/Project/test1.py
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-#from transformers import BertTokenizer, BertForSequenceClassification
 import math, keras_ocr
 # Initialize pipeline
 pipeline = None
 model_path="NSFW_text_classifier"
-#tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-#model_2 = BertForSequenceClassification.from_pretrained("CustomModel")
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForSequenceClassification.from_pretrained(model_path)
-#model_2.to('cpu')
+
 import streamlit as st
 
 def get_distance(predictions):
@@ -107,10 +104,6 @@ if uploaded_file is not None:
 
     input_text =sentance
     print(input_text)
-    #inputs = tokenizer(text,padding = True, truncation = True, return_tensors='pt').to('cpu')
-    #outputs = model_2(**inputs)
-    #predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
-    #predictions = predictions.cpu().detach().numpy()
     inputs = tokenizer(input_text, return_tensors="pt")
     outputs = model(**inputs)
     predictions = outputs.logits.softmax(dim=-1)