Spaces:
Runtime error
Runtime error
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 <fffiloni@users.noreply.huggingface.co>
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 |
|