Update app.py

app.py

@@ -19,10 +19,10 @@ import io
 from transformers import AutoTokenizer, AutoModel
 from utils import best_text_choice
 
-tokenizer = AutoTokenizer.from_pretrained("ai-forever/ru-en-RoSBERTa")
-model = AutoModel.from_pretrained("ai-forever/ru-en-RoSBERTa")
+tokenizer = AutoTokenizer.from_pretrained("Sergeyzh/rubert-tiny-turbo")
+model = AutoModel.from_pretrained("Sergeyzh/rubert-tiny-turbo")
 # Load the DataFrame used in the best_text_choice function
-unique_sms_df = pd.read_parquet('unique_texts.parquet')
+unique_sms_df = pd.read_parquet('embeddings_tiny_bert.parquet')
 
 MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
 token = os.getenv('GITHUB_TOKEN')