Madras1 commited on
Commit
dbd95ef
·
verified ·
1 Parent(s): e1fffbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -14
app.py CHANGED
@@ -165,19 +165,15 @@ def router(message, history, model_selector, request: gr.Request):
165
  else:
166
  messages.append({"role": "user", "content": str(message)})
167
 
168
- # Seleção de Modelos (Atualizada com seus pedidos)
169
- if any(x in model_selector for x in ["Gemini", "LearnLM", "Gemma"]):
170
  tid = "gemini-1.5-flash"
171
 
172
- # Mapeamento estrito dos modelos novos
173
- if "LearnLM 2.0" in model_selector: tid = "learnlm-2.0-flash-experimental"
174
- elif "Gemma 3" in model_selector: tid = "gemma-3-27b"
175
- elif "2.5 Lite" in model_selector: tid = "gemini-2.5-flash-lite"
176
-
177
- # Mapeamento dos anteriores
178
  elif "3.0" in model_selector: tid = "gemini-3.0-pro-preview"
179
  elif "2.5 Pro" in model_selector: tid = "gemini-2.5-pro"
180
- elif "2.5 Flash" in model_selector: tid = "gemini-2.5-flash" # Cuidado para não confundir com Lite
181
  elif "2.0" in model_selector: tid = "gemini-2.0-flash-exp"
182
 
183
  return run_gemini(messages, tid)
@@ -202,14 +198,11 @@ def router(message, history, model_selector, request: gr.Request):
202
  return "⚠️ Modelo não reconhecido."
203
 
204
  # --- 6. INTERFACE ---
205
- # Berta removeu o theme=... aqui para não dar erro
206
  with gr.Blocks() as demo:
207
- gr.Markdown("# 🔀 APIDOST v11 - Fixed & Expanded")
208
 
209
  models_list = [
210
- "✨ Google: LearnLM 2.0 Flash (Exp) 📚", # ID: learnlm-2.0-flash-experimental
211
- "✨ Google: Gemma 3 27B 💎", # ID: gemma-3-27b
212
- "✨ Google: Gemini 2.5 Flash Lite ⚡", # ID: gemini-2.5-flash-lite
213
  "✨ Google: Gemini 3.0 Pro (Experimental)",
214
  "✨ Google: Gemini 2.5 Pro",
215
  "✨ Google: Gemini 2.5 Flash",
 
165
  else:
166
  messages.append({"role": "user", "content": str(message)})
167
 
168
+ # Seleção de Modelos (Limpa e corrigida)
169
+ if "Gemini" in model_selector:
170
  tid = "gemini-1.5-flash"
171
 
172
+ # Mapeamento
173
+ if "2.5 Lite" in model_selector: tid = "gemini-2.5-flash-lite"
 
 
 
 
174
  elif "3.0" in model_selector: tid = "gemini-3.0-pro-preview"
175
  elif "2.5 Pro" in model_selector: tid = "gemini-2.5-pro"
176
+ elif "2.5 Flash" in model_selector: tid = "gemini-2.5-flash"
177
  elif "2.0" in model_selector: tid = "gemini-2.0-flash-exp"
178
 
179
  return run_gemini(messages, tid)
 
198
  return "⚠️ Modelo não reconhecido."
199
 
200
  # --- 6. INTERFACE ---
 
201
  with gr.Blocks() as demo:
202
+ gr.Markdown("# 🔀 APIDOST v12 - Estável")
203
 
204
  models_list = [
205
+ "✨ Google: Gemini 2.5 Flash Lite ",
 
 
206
  "✨ Google: Gemini 3.0 Pro (Experimental)",
207
  "✨ Google: Gemini 2.5 Pro",
208
  "✨ Google: Gemini 2.5 Flash",