Spaces:

Bijoy09
/

Bangla_spam_sms_detection_app

Running

Bijoy09 commited on 10 days ago

Commit

384e68d

•

1 Parent(s): cd6d890

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ app = FastAPI()
 os.environ['TRANSFORMERS_CACHE'] = os.getenv('TRANSFORMERS_CACHE', '/app/cache')
 # Load model and tokenizer
-model_name = "Bijoy09/Bangla_spam_sms_detection_app"
 try:
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)

 os.environ['TRANSFORMERS_CACHE'] = os.getenv('TRANSFORMERS_CACHE', '/app/cache')
 # Load model and tokenizer
+model_name = "Bijoy09/MObilebert"
 try:
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)