Spaces:
Running
on
Zero
Running
on
Zero
quantumiracle
commited on
Commit
·
01f7020
1
Parent(s):
6367b1e
fix
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import os
|
2 |
import gradio as gr
|
3 |
from gradio_imageslider import ImageSlider
|
@@ -11,7 +12,6 @@ from CKPT_PTH import LLAVA_MODEL_PATH
|
|
11 |
import einops
|
12 |
import copy
|
13 |
import time
|
14 |
-
import spaces
|
15 |
|
16 |
# Initialize devices
|
17 |
if torch.cuda.device_count() >= 2:
|
|
|
1 |
+
import spaces
|
2 |
import os
|
3 |
import gradio as gr
|
4 |
from gradio_imageslider import ImageSlider
|
|
|
12 |
import einops
|
13 |
import copy
|
14 |
import time
|
|
|
15 |
|
16 |
# Initialize devices
|
17 |
if torch.cuda.device_count() >= 2:
|