Spaces:

arjunanand13
/

video_ad_classifier

Sleeping

arjunanand13 commited on Aug 30, 2024

Commit

d0774b9

verified ·

1 Parent(s): 43d9b67

Update image_caption.py

Files changed (1) hide show

image_caption.py CHANGED Viewed

@@ -12,7 +12,7 @@ class Caption:
         self.api_key = 'AIzaSyBPQF0g5EfEPzEiGRzA3iNzJZK4jDukMvE'
         genai.configure(api_key=self.api_key)
-        self.model = genai.GenerativeModel(model_name="gemini-pro-vision")
     #     self.model = VisionEncoderDecoderModel.from_pretrained(
     #         "nlpconnect/vit-gpt2-image-captioning"
     #     )

         self.api_key = 'AIzaSyBPQF0g5EfEPzEiGRzA3iNzJZK4jDukMvE'
         genai.configure(api_key=self.api_key)
+        self.model = genai.GenerativeModel(model_name="gemini-1.5-flash")
     #     self.model = VisionEncoderDecoderModel.from_pretrained(
     #         "nlpconnect/vit-gpt2-image-captioning"
     #     )