Update app.py

app.py

@@ -7,20 +7,13 @@ import json
 import shutil, os
 
 offload_folder = "offload/"
-if os.path.exists(offload_folder):
-    shutil.rmtree(offload_folder)
-os.makedirs(offload_folder)
+os.makedirs(offload_folder, exist_ok=True)
 
 model_id = "deepseek-ai/deepseek-coder-1.3b-base"
 lora_id = "Seunggg/lora-plant"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 
-offload_folder = "offload/"
-if os.path.exists(offload_folder):
-    shutil.rmtree(offload_folder)
-os.makedirs(offload_folder)
-
 base = AutoModelForCausalLM.from_pretrained(
     model_id,
     device_map="auto",