allbibek commited on
Commit
1ea1920
·
verified ·
1 Parent(s): 20c2370

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -21,6 +21,15 @@ else:
21
  embedder = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
22
  reranker = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
23
 
 
 
 
 
 
 
 
 
 
24
  # === Supabase ===
25
  supabase_url = os.getenv("SUPABASE_URL")
26
  supabase_key = os.getenv("SUPABASE_KEY")
@@ -59,9 +68,9 @@ def expand_query(query: str, num_variations: int = 3) -> str:
59
  try:
60
  # Panggil Gemini
61
  response = gemini_client.models.generate_content(
62
- # model="gemini-2.5-flash",
63
  # model="gemini-robotics-er-1.5-preview",
64
- model="gemini-3.1-flash-lite",
65
  contents=prompt,
66
  )
67
 
 
21
  embedder = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
22
  reranker = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
23
 
24
+ print("--- Daftar Model yang Tersedia untuk Kamu ---")
25
+
26
+ # 2. Ambil semua model yang bisa digunakan untuk teks (generateContent)
27
+ for m in genai.list_models():
28
+ if 'generateContent' in m.supported_generation_methods:
29
+ print(f"ID Model: {m.name} | Kuota: {m.display_name}")
30
+
31
+ print("---------------------------------------------")
32
+
33
  # === Supabase ===
34
  supabase_url = os.getenv("SUPABASE_URL")
35
  supabase_key = os.getenv("SUPABASE_KEY")
 
68
  try:
69
  # Panggil Gemini
70
  response = gemini_client.models.generate_content(
71
+ model="gemini-2.5-flash",
72
  # model="gemini-robotics-er-1.5-preview",
73
+ # model="gemini-3.1-flash-lite",
74
  contents=prompt,
75
  )
76