Spaces:

emr-distillation
/

Onpremise_LLM_Normal_Detection

Sleeping

emr-distillation commited on Feb 18, 2024

Commit

81c3b9b

verified ·

1 Parent(s): 3da59a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import re
 def paragraph_leveling(text):
-    model_name = "./trained_model/fine_tunning_encoder_v2"
     model = AutoModel.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained('zzxslp/RadBERT-RoBERTa-4m')

 def paragraph_leveling(text):
+    model_name = "./trained_model/contrastive_encoder_sentence"
     model = AutoModel.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained('zzxslp/RadBERT-RoBERTa-4m')