Aashi fffiloni commited on
Commit
f6ae4ba
1 Parent(s): 81f232f

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

Files changed (1) hide show
  1. app.py +1 -0
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