Update app.py

app.py

@@ -5,11 +5,11 @@ def generate_text(context, num_samples, context_length, model_name):
     from pathlib import Path
     
     if model_name == "pythia_160m_deduped_custom" or model_name == "pythia_160m_deduped_huggingface":
-        ckpt_dir = Path('checkpoints/EleutherAI/pythia-160m-deduped')
+        ckpt_dir = Path('./checkpoints/EleutherAI/pythia-160m-deduped')
     elif model_name == "pythia_70m_deduped":
-        ckpt_dir = Path('checkpoints/EleutherAI/pythia-70m-deduped')
+        ckpt_dir = Path('./checkpoints/EleutherAI/pythia-70m-deduped')
     elif model_name == "pythia_410m_deduped":
-        ckpt_dir = Path('checkpoints/EleutherAI/pythia-410m-deduped')        
+        ckpt_dir = Path('./checkpoints/EleutherAI/pythia-410m-deduped')        
         
     context = str(context)
     num_samples = int(num_samples)