Merge branch 'main' of https://huggingface.co/spaces/pablo-sampaio/futeboy into main

app.py

@@ -223,7 +223,8 @@ else:
     HF_KEY = os.environ['HUGGINGFACE_API_KEY']
 
     # Serverless API endpoint for OpenAI's Whisper model
-    ASR_API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v3"
+    #ASR_API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v3"
+    ASR_API_URL = "https://api-inference.huggingface.co/models/openai/whisper-small"
     ASR_API_HEADERS = {"Authorization": f"Bearer {HF_KEY}"}
 
 

match_info_crawler.py

@@ -20,7 +20,8 @@ load_dotenv(find_dotenv()) # should load the GOOGLE_API_KEY
 genai.configure()
 
 
-USE_BEST_EXTRACTOR = True
+MODEL_VERSION = "gemini-1.5-flash"
+#MODEL_VERSION = "gemini-1.0-pro"
 
 sao_paulo_timezone = pytz.timezone('America/Sao_Paulo')
 
@@ -45,11 +46,11 @@ Palmeiras 0 x 2 Atletico Paranaense - Campeonato Brasileiro - Série A - finishe
 ```
 '''
 
-if USE_BEST_EXTRACTOR:
-    EXTRACTOR_MODEL = genai.GenerativeModel('gemini-1.5-pro-latest', 
-                               system_instruction=SYSTEM_PROMPT)  # TODO: setar uma temperatura bem baixa!
+if MODEL_VERSION != "gemini-1.0-pro":
+    EXTRACTOR_MODEL = genai.GenerativeModel(MODEL_VERSION, 
+                                            system_instruction=SYSTEM_PROMPT)  # TODO: setar uma temperatura bem baixa!
 else:
-    EXTRACTOR_MODEL = genai.GenerativeModel('gemini-1.0-pro')
+    EXTRACTOR_MODEL = genai.GenerativeModel(MODEL_VERSION)
 
 
 START_MAIS_POP_SECTION = "<h3 class='match-list_league-name'>MAIS POPULARES AGORA</h3>"
@@ -122,7 +123,7 @@ def get_matches_info(date_str: str):
     #print("PROCESSING:\n", plain_page)
 
     try:
-        if USE_BEST_EXTRACTOR:
+        if MODEL_VERSION != "gemini-1.0-pro":
             #response = EXTRACTOR_MODEL.generate_content(plain_page)
             response = EXTRACTOR_MODEL.generate_content("HTML PAGE TO BE PARSED: \n\n" + plain_page)
         else: