fruitpicker01 commited on
Commit
5358237
1 Parent(s): 1faa1c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,10 +22,10 @@ from utils import best_text_choice
22
  import httpx
23
  import asyncio
24
 
25
- tokenizer = AutoTokenizer.from_pretrained("ai-forever/ru-en-RoSBERTa")
26
- model = AutoModel.from_pretrained("ai-forever/ru-en-RoSBERTa")
27
  # Load the DataFrame used in the best_text_choice function
28
- unique_sms_df = pd.read_parquet('unique_texts.parquet')
29
 
30
  MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
31
  token = os.getenv('GITHUB_TOKEN')
 
22
  import httpx
23
  import asyncio
24
 
25
+ tokenizer = AutoTokenizer.from_pretrained("Sergeyzh/rubert-tiny-turbo")
26
+ model = AutoModel.from_pretrained("Sergeyzh/rubert-tiny-turbo")
27
  # Load the DataFrame used in the best_text_choice function
28
+ unique_sms_df = pd.read_parquet('embeddings_tiny_bert.parquet')
29
 
30
  MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
31
  token = os.getenv('GITHUB_TOKEN')