Reham1 commited on
Commit
952d777
·
verified ·
1 Parent(s): 38fcedb

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -12,7 +12,7 @@ def load_model():
12
  "Reham1/mistral-diabetes-lora1",
13
  torch_dtype="auto",
14
  device_map="auto",
15
- offload_folder="/tmp/offload", # ✅ تفادي مشاكل الصلاحيات
16
  token=API_TOKEN
17
  )
18
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
12
  "Reham1/mistral-diabetes-lora1",
13
  torch_dtype="auto",
14
  device_map="auto",
15
+ offload_folder="/tmp/offload",
16
  token=API_TOKEN
17
  )
18
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)