Nymbo commited on
Commit
272be4b
1 Parent(s): 7c345ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
@@ -139,7 +139,7 @@ image_adapter.eval()
139
  image_adapter.to("cuda")
140
 
141
 
142
- @spaces.GPU()
143
  @torch.no_grad()
144
  def stream_chat(input_image: Image.Image, caption_type: str, caption_tone: str, caption_length: str | int) -> str:
145
  torch.cuda.empty_cache()
 
1
+ # import spaces
2
  import gradio as gr
3
  from huggingface_hub import InferenceClient
4
  from torch import nn
 
139
  image_adapter.to("cuda")
140
 
141
 
142
+ # @spaces.GPU()
143
  @torch.no_grad()
144
  def stream_chat(input_image: Image.Image, caption_type: str, caption_tone: str, caption_length: str | int) -> str:
145
  torch.cuda.empty_cache()