anilbhatt1 commited on
Commit
6b9015d
1 Parent(s): 0f48d1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,11 +5,11 @@ def generate_text(context, num_samples, context_length, model_name):
5
  from pathlib import Path
6
 
7
  if model_name == "pythia_160m_deduped_custom" or model_name == "pythia_160m_deduped_huggingface":
8
- ckpt_dir = Path('./checkpoints/EleutherAI/pythia-160m-deduped')
9
  elif model_name == "pythia_70m_deduped":
10
- ckpt_dir = Path('./checkpoints/EleutherAI/pythia-70m-deduped')
11
  elif model_name == "pythia_410m_deduped":
12
- ckpt_dir = Path('./checkpoints/EleutherAI/pythia-410m-deduped')
13
 
14
  context = str(context)
15
  num_samples = int(num_samples)
 
5
  from pathlib import Path
6
 
7
  if model_name == "pythia_160m_deduped_custom" or model_name == "pythia_160m_deduped_huggingface":
8
+ ckpt_dir = Path('/home/user/app/checkpoints/EleutherAI/pythia-160m-deduped')
9
  elif model_name == "pythia_70m_deduped":
10
+ ckpt_dir = Path('/home/user/app/checkpoints/EleutherAI/pythia-70m-deduped')
11
  elif model_name == "pythia_410m_deduped":
12
+ ckpt_dir = Path('/home/user/app/checkpoints/EleutherAI/pythia-410m-deduped')
13
 
14
  context = str(context)
15
  num_samples = int(num_samples)