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 |
|