PearlIsa commited on
Commit
897c7a4
·
verified ·
1 Parent(s): e19db79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -36,13 +36,8 @@ hf_token = os.getenv("HF_TOKEN")
36
  if hf_token:
37
  login(token=hf_token)
38
 
39
- # Set environment variables programmatically
40
- os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
41
- os.environ["TRANSFORMERS_OFFLINE"] = "1"
42
 
43
- # You can also add these configurations:
44
- os.environ["CUDA_VISIBLE_DEVICES"] = "0" # Use first GPU
45
- os.environ["TORCH_CUDA_ARCH_LIST"] = "7.5" # Optimize for common GPU architectures
46
 
47
  class AdaptiveMedicalBot:
48
  def __init__(self):
@@ -609,6 +604,7 @@ if __name__ == "__main__":
609
  login(token=hf_token)
610
 
611
  # Launch demo
 
612
  demo = create_demo()
613
  demo.launch(share=True)
614
 
 
36
  if hf_token:
37
  login(token=hf_token)
38
 
39
+ os.environ.pop("HF_HUB_OFFLINE", None)
 
 
40
 
 
 
 
41
 
42
  class AdaptiveMedicalBot:
43
  def __init__(self):
 
604
  login(token=hf_token)
605
 
606
  # Launch demo
607
+ os.environ.pop("HF_HUB_OFFLINE", None) # Ensure online mode
608
  demo = create_demo()
609
  demo.launch(share=True)
610