Update app.py
Browse files
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 |
|