fffiloni commited on
Commit
ad5a806
1 Parent(s): 9da309a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  from gradio_client import Client
3
  import cv2
@@ -91,6 +92,7 @@ model = AutoModelForCausalLM.from_pretrained(
91
  )
92
  tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
93
 
 
94
  def process_image(image_in):
95
  '''
96
  client = Client("https://vikhyatk-moondream1.hf.space/")
@@ -129,6 +131,7 @@ def get_salmonn(audio_in):
129
  print(result)
130
  return result
131
 
 
132
  def llm_process(user_prompt):
133
  agent_maker_sys = standard_sys
134
 
 
1
+ import spaces
2
  import gradio as gr
3
  from gradio_client import Client
4
  import cv2
 
92
  )
93
  tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
94
 
95
+ @spaces.GPU()
96
  def process_image(image_in):
97
  '''
98
  client = Client("https://vikhyatk-moondream1.hf.space/")
 
131
  print(result)
132
  return result
133
 
134
+ @spaces.GPU()
135
  def llm_process(user_prompt):
136
  agent_maker_sys = standard_sys
137