Commit
•
f8c52ba
1
Parent(s):
eb70836
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
from huggingface_hub import InferenceClient
|
| 4 |
from torch import nn
|
|
@@ -61,7 +61,7 @@ image_adapter.eval()
|
|
| 61 |
image_adapter.to("cuda")
|
| 62 |
|
| 63 |
|
| 64 |
-
|
| 65 |
@torch.no_grad()
|
| 66 |
def stream_chat(input_image: Image.Image):
|
| 67 |
torch.cuda.empty_cache()
|
|
|
|
| 1 |
+
#import spaces
|
| 2 |
import gradio as gr
|
| 3 |
from huggingface_hub import InferenceClient
|
| 4 |
from torch import nn
|
|
|
|
| 61 |
image_adapter.to("cuda")
|
| 62 |
|
| 63 |
|
| 64 |
+
#@spaces.GPU()
|
| 65 |
@torch.no_grad()
|
| 66 |
def stream_chat(input_image: Image.Image):
|
| 67 |
torch.cuda.empty_cache()
|