Pclanglais commited on
Commit
5675a74
·
verified ·
1 Parent(s): c9e2313

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-Rag"
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')