AI-Anchorite commited on
Commit
cf97cc7
·
verified ·
1 Parent(s): 8a4f7ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
@@ -156,8 +156,8 @@ image_adapter.eval()
156
  image_adapter.to("cuda")
157
 
158
 
159
- @spaces.GPU()
160
- @torch.no_grad()
161
  def stream_chat(input_image: Image.Image, caption_type: str, caption_length: str | int, extra_options: list[str], name_input: str, custom_prompt: str) -> tuple[str, str]:
162
  torch.cuda.empty_cache()
163
 
 
1
+ #import spaces
2
  import gradio as gr
3
  from huggingface_hub import InferenceClient
4
  from torch import nn
 
156
  image_adapter.to("cuda")
157
 
158
 
159
+ #@spaces.GPU()
160
+ #@torch.no_grad()
161
  def stream_chat(input_image: Image.Image, caption_type: str, caption_length: str | int, extra_options: list[str], name_input: str, custom_prompt: str) -> tuple[str, str]:
162
  torch.cuda.empty_cache()
163