rishiraj commited on
Commit
835e6a5
1 Parent(s): 6126c75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ with open(".config/application_default_credentials.json", 'w') as file:
14
 
15
  vertexai.init(project=os.getenv('project_id'))
16
  model = GenerativeModel("gemini-1.0-pro-vision")
17
- client = InferenceClient("unsloth/gemma-7b-it-bnb-4bit")
18
 
19
  def extract_image_urls(text):
20
  url_regex = r"(https?:\/\/.*\.(?:png|jpg|jpeg|gif|webp|svg))"
 
14
 
15
  vertexai.init(project=os.getenv('project_id'))
16
  model = GenerativeModel("gemini-1.0-pro-vision")
17
+ client = InferenceClient("google/gemma-7b")
18
 
19
  def extract_image_urls(text):
20
  url_regex = r"(https?:\/\/.*\.(?:png|jpg|jpeg|gif|webp|svg))"