mjavaid commited on
Commit
278a9c4
·
1 Parent(s): d58a265

first commit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def load_model():
23
  except Exception as e:
24
  return f"Error loading model: {e}\nMake sure to install the correct version of transformers with: pip install 'git+https://github.com/huggingface/[email protected]'"
25
  return "Model already loaded!"
26
- @spaces.gpu
27
  def process_image_and_prompt(image, image_url, prompt, temperature=0.3, max_tokens=300):
28
  global processor, model
29
 
 
23
  except Exception as e:
24
  return f"Error loading model: {e}\nMake sure to install the correct version of transformers with: pip install 'git+https://github.com/huggingface/[email protected]'"
25
  return "Model already loaded!"
26
+ @spaces.GPU
27
  def process_image_and_prompt(image, image_url, prompt, temperature=0.3, max_tokens=300):
28
  global processor, model
29