add whisper

app.py

@@ -4,7 +4,7 @@ import os
 
 from faster_whisper import WhisperModel
 # tiny, tiny.en, base, base.en, small, small.en, medium, medium.en, large-v1, large-v2, large-v3, or large
-model_name = 'tiny.en'
+model_name = 'base'
 model = WhisperModel(model_name, device="cpu", download_root="./models")
 
 ydl_opts = {

utils.py

@@ -0,0 +1,43 @@
+import google.generativeai as genai
+
+API_KEY = "AIzaSyCkqv9dWrlbRjv9fHO_O8jBORGfYVPJTnY"
+
+def call_gemini(prompt="", given_text=None, given_image=None, generation_config=None, safety_settings=None):
+    genai.configure(api_key=API_KEY)
+    generation_config = {
+            "temperature": 0.8,
+            "top_p": 1,
+            "top_k": 32,
+            "max_output_tokens": 8192,
+    }
+
+    safety_settings = [
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+        ]
+    
+    model = genai.GenerativeModel(model_name='gemini-pro',
+                                generation_config=generation_config,
+                                safety_settings=safety_settings)
+    
+    prompt_parts = "如何学习rust语言?"
+
+    response = model.generate_content(prompt_parts)
+    print(response.text)
+
+if __name__ == "__main__":
+    call_gemini()