rinrikatoki commited on
Commit
4d73c77
·
verified ·
1 Parent(s): c8b0b86

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -7
app.py CHANGED
@@ -13,17 +13,26 @@ if not hf_token:
13
  # 🔹 لاگین به HuggingFace
14
  login(hf_token)
15
 
16
- # 🔹 تنظیم مسیرها
17
- LORA_ZIP_PATH = "dorna-diabetes-finetuned-20250514T183411Z-1-001.zip" # نام فایل زیپ که آپلود کردی
18
- LORA_PATH = "./dorna-diabetes-finetuned"
19
 
20
- # ✅ اکسترکت کردن فایل LoRA
21
- if not os.path.exists(LORA_PATH):
22
  with zipfile.ZipFile(LORA_ZIP_PATH, 'r') as zip_ref:
23
- zip_ref.extractall(LORA_PATH)
24
  print("✅ فایل LoRA اکسترکت شد.")
25
 
26
- # ✅ تغییر نام فایل .safetensors به adapter_model.safetensors
 
 
 
 
 
 
 
 
 
27
  for filename in os.listdir(LORA_PATH):
28
  if filename.endswith(".safetensors") and filename != "adapter_model.safetensors":
29
  os.rename(
 
13
  # 🔹 لاگین به HuggingFace
14
  login(hf_token)
15
 
16
+ # 🔹 مسیر فایل زیپ
17
+ LORA_ZIP_PATH = "dorna-diabetes-finetuned-20250514T183411Z-1-001.zip"
18
+ EXTRACT_DIR = "lora" # پوشه‌ای که فایل‌ها داخلش اکسترکت می‌شن
19
 
20
+ # ✅ اکسترکت فایل ZIP در صورت نیاز
21
+ if not os.path.exists(EXTRACT_DIR):
22
  with zipfile.ZipFile(LORA_ZIP_PATH, 'r') as zip_ref:
23
+ zip_ref.extractall(EXTRACT_DIR)
24
  print("✅ فایل LoRA اکسترکت شد.")
25
 
26
+ # ✅ پیدا کردن مسیر فولدر واقعی شامل adapter_config.json
27
+ # معمولاً zip یک زیرپوشه می‌سازه داخل خودش
28
+ for root, dirs, files in os.walk(EXTRACT_DIR):
29
+ if "adapter_config.json" in files:
30
+ LORA_PATH = root
31
+ break
32
+ else:
33
+ raise FileNotFoundError("❌ adapter_config.json در هیچ زیرپوشه‌ای یافت نشد.")
34
+
35
+ # ✅ تغییر نام فایل .safetensors به adapter_model.safetensors اگر لازم بود
36
  for filename in os.listdir(LORA_PATH):
37
  if filename.endswith(".safetensors") and filename != "adapter_model.safetensors":
38
  os.rename(