aliasgerovs commited on
Commit
2e5c3b9
1 Parent(s): e611170
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -158,15 +158,11 @@ AI DETECTION SECTION
158
 
159
  text_bc_model_path = "polygraf-ai/ai-text-bc-bert-1-4m"
160
  text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
161
- text_bc_model = AutoModelForSequenceClassification.from_pretrained(
162
- text_bc_model_path
163
- )
164
 
165
  text_mc_model_path = "polygraf-ai/ai-text-mc-v5-lighter-spec"
166
  text_mc_tokenizer = AutoTokenizer.from_pretrained(text_mc_model_path)
167
- text_mc_model = AutoModelForSequenceClassification.from_pretrained(
168
- text_mc_model_path
169
- )
170
 
171
  def remove_special_characters(text):
172
  cleaned_text = re.sub(r'[^a-zA-Z0-9\s]', '', text)
 
158
 
159
  text_bc_model_path = "polygraf-ai/ai-text-bc-bert-1-4m"
160
  text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
161
+ text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path)
 
 
162
 
163
  text_mc_model_path = "polygraf-ai/ai-text-mc-v5-lighter-spec"
164
  text_mc_tokenizer = AutoTokenizer.from_pretrained(text_mc_model_path)
165
+ text_mc_model = AutoModelForSequenceClassification.from_pretrained(text_mc_model_path)
 
 
166
 
167
  def remove_special_characters(text):
168
  cleaned_text = re.sub(r'[^a-zA-Z0-9\s]', '', text)