Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ import pandas as pd
|
|
13 |
# Define the device
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
15 |
|
16 |
-
model_name = "PleIAs/Pleias-
|
17 |
|
18 |
# Get Hugging Face token from environment variable
|
19 |
hf_token = os.environ.get('HF_TOKEN')
|
|
|
13 |
# Define the device
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
15 |
|
16 |
+
model_name = "PleIAs/Pleias-3b-rag"
|
17 |
|
18 |
# Get Hugging Face token from environment variable
|
19 |
hf_token = os.environ.get('HF_TOKEN')
|