Update app.py

app.py

@@ -10,7 +10,7 @@ checkpoint = "tiiuae/falcon-7b"
 # checkpoint = "vsrinivas/FalconLite2"
 
 model = AutoModelForCausalLM.from_pretrained(
-    checkpoint, device_map="auto", offload_folder="offload", 
+    checkpoint, device_map="auto", offload_folder="https://huggingface.co/vsrinivas/offload", 
     trust_remote_code=True,
     # torch_dtype="auto",
 )
@@ -33,7 +33,7 @@ pipeline = transformers.pipeline(
     # torch_dtype=torch.bfloat16,
     # trust_remote_code=True,
     device_map="auto",
-    offload_folder="offload",
+    offload_folder="https://huggingface.co/vsrinivas/offload",
     offload_state_dict = True,
 )