han1997 commited on
Commit
7fcce9b
·
verified ·
1 Parent(s): e3bcb16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,6 @@ import sys
14
  import subprocess
15
  from PIL import Image
16
 
17
- from cobra import load
18
  import time
19
 
20
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'packaging'])
@@ -22,6 +21,7 @@ subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'ninja'])
22
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'mamba-ssm'])
23
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'causal-conv1d'])
24
 
 
25
  vlm = load("cobra+3b")
26
 
27
  if torch.cuda.is_available():
 
14
  import subprocess
15
  from PIL import Image
16
 
 
17
  import time
18
 
19
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'packaging'])
 
21
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'mamba-ssm'])
22
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'causal-conv1d'])
23
 
24
+ from cobra import load
25
  vlm = load("cobra+3b")
26
 
27
  if torch.cuda.is_available():