added roberta again

app.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import torch
-from transformers import BertTokenizer, BertForSequenceClassification
+from transformers import RobertaTokenizer, RobertaForSequenceClassification
 from torch.nn.functional import softmax
 import re
 
@@ -15,8 +15,8 @@ app = FastAPI(
 class ContactDetector:
     def __init__(self):
         cache_dir = "/app/model_cache"
-        self.tokenizer = BertTokenizer.from_pretrained('bert-large-uncased', cache_dir=cache_dir)
-        self.model = BertForSequenceClassification.from_pretrained('bert-large-uncased', num_labels=2, cache_dir=cache_dir)
+        self.tokenizer = RobertaTokenizer.from_pretrained('roberta-base', cache_dir=cache_dir)
+        self.model = RobertaForSequenceClassification.from_pretrained('roberta-base', num_labels=2, cache_dir=cache_dir)
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.model.to(self.device)
         self.model.eval()