Mountchicken commited on
Commit
7e3dc12
·
verified ·
1 Parent(s): bf00d99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -4,6 +4,8 @@ import os
4
  import sys
5
  import threading
6
 
 
 
7
 
8
  # Add pre-build setup for Hugging Face Spaces
9
  def setup_environment():
@@ -22,6 +24,7 @@ def setup_environment():
22
 
23
  # Run setup
24
  setup_environment()
 
25
 
26
  import gradio as gr
27
  import numpy as np
@@ -80,6 +83,7 @@ def initialize_models(args):
80
  return (gdino_model, processor, model)
81
 
82
 
 
83
  def process_image_with_streaming(
84
  image,
85
  system_prompt,
@@ -199,6 +203,7 @@ def process_image_with_streaming(
199
  return gdino_vis_result, ref_vis_result, output_text
200
 
201
 
 
202
  def process_image_non_streaming(
203
  image,
204
  system_prompt,
@@ -225,6 +230,7 @@ def process_image_non_streaming(
225
  )
226
 
227
 
 
228
  def create_streaming_interface(models):
229
  """Create a streaming interface using a different approach"""
230
  (
 
4
  import sys
5
  import threading
6
 
7
+ import spaces
8
+
9
 
10
  # Add pre-build setup for Hugging Face Spaces
11
  def setup_environment():
 
24
 
25
  # Run setup
26
  setup_environment()
27
+ os.system("pip install -r requirements_hf.txt")
28
 
29
  import gradio as gr
30
  import numpy as np
 
83
  return (gdino_model, processor, model)
84
 
85
 
86
+ @spaces.GPU
87
  def process_image_with_streaming(
88
  image,
89
  system_prompt,
 
203
  return gdino_vis_result, ref_vis_result, output_text
204
 
205
 
206
+ @spaces.GPU
207
  def process_image_non_streaming(
208
  image,
209
  system_prompt,
 
230
  )
231
 
232
 
233
+ @spaces.GPU
234
  def create_streaming_interface(models):
235
  """Create a streaming interface using a different approach"""
236
  (