Merge branch 'main' of https://huggingface.co/spaces/RaviNaik/ERA-SESSION22

app.py

@@ -12,7 +12,7 @@ def generate(prompt, max_new_tokens, temperature, num_samples):
 
     responses = main(
         prompt=prompt,
-        checkpoint_dir=Path("out/redpajama"),
+        checkpoint_dir=Path("./out/redpajama"),
         max_new_tokens=max_new_tokens,
         temperature=temperature,
         num_samples=num_samples,