fruitpicker01
commited on
Commit
•
5358237
1
Parent(s):
1faa1c5
Update app.py
Browse files
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("
|
26 |
-
model = AutoModel.from_pretrained("
|
27 |
# Load the DataFrame used in the best_text_choice function
|
28 |
-
unique_sms_df = pd.read_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')
|