Spaces:
Running
on
Zero
Running
on
Zero
decorate the right function for ZERO gpu inference (#1)
Browse files- decorate the right function for ZERO gpu inference (9135ce43958bb849c200dea82e3c02e77c0dde9c)
Co-authored-by: Sylvain Filoni <[email protected]>
app.py
CHANGED
@@ -21,6 +21,7 @@ def load_llama_model():
|
|
21 |
|
22 |
@spaces.GPU
|
23 |
# Function to generate predictions for text and image
|
|
|
24 |
def process_input(text, image=None):
|
25 |
model, processor = load_llama_model()
|
26 |
|
|
|
21 |
|
22 |
@spaces.GPU
|
23 |
# Function to generate predictions for text and image
|
24 |
+
@spaces.GPU
|
25 |
def process_input(text, image=None):
|
26 |
model, processor = load_llama_model()
|
27 |
|