arjunanand13 commited on
Commit
ff55b77
·
verified ·
1 Parent(s): d0774b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ class VideoClassifier:
49
  def setup_gemini_model(self):
50
  self.genai = genai
51
  self.genai.configure(api_key="AIzaSyBPQF0g5EfEPzEiGRzA3iNzJZK4jDukMvE")
52
- self.genai_model = genai.GenerativeModel('gemini-1.5-flash')
53
  self.whisper_model = whisper.load_model("base")
54
  self.img_cap = Caption()
55
 
 
49
  def setup_gemini_model(self):
50
  self.genai = genai
51
  self.genai.configure(api_key="AIzaSyBPQF0g5EfEPzEiGRzA3iNzJZK4jDukMvE")
52
+ self.genai_model = genai.GenerativeModel('gemini-pro')
53
  self.whisper_model = whisper.load_model("base")
54
  self.img_cap = Caption()
55