frendyrachman commited on
Commit
caf5e05
·
verified ·
1 Parent(s): 400217f

Update app.py

Browse files

Change model to gemini 2.0 lite

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -131,7 +131,7 @@ Berikut kami informasikan kekurangan dokumen yang *WAJIB* dibawa saat biometric
131
 
132
  # Perform document analysis
133
  response_1 = client.models.generate_content(
134
- model="gemini-2.0-flash",
135
  contents=[prompt] + images,
136
  config=GenerateContentConfig(
137
  temperature=0.2,
@@ -142,7 +142,7 @@ Berikut kami informasikan kekurangan dokumen yang *WAJIB* dibawa saat biometric
142
  # Perform valid list generation
143
  prompt_2 = "Dari analisa diatas, Rangkum menjadi list/daftar dokumen yang valid saja. Misal: 'paspor', 'pasfoto', 'kartukeluarga', 'buktinikah', 'KTP', 'suratkelahiran', 'suratsponsor', 'suratkerja', 'NIB', 'SIUP', 'suratjaminanstaff', 'suratsekolah', 'kontrakkerja', 'suratpensiun', 'rekeningkoran', 'slipgaji', 'other'. Jangan tambahkan kata-kata atau kalimat lain."
144
  response_2 = client.models.generate_content(
145
- model="gemini-2.0-flash",
146
  contents=[analysis, prompt_2], # Pass prompt and image files
147
  config=GenerateContentConfig(
148
  temperature=0.2,
@@ -186,7 +186,7 @@ Output:
186
  renamed_images = []
187
  for image in images:
188
  response = client.models.generate_content(
189
- model="gemini-2.0-flash",
190
  contents=[prompt_3, image],
191
  config=GenerateContentConfig(
192
  temperature=0.2,
 
131
 
132
  # Perform document analysis
133
  response_1 = client.models.generate_content(
134
+ model="gemini-2.0-flash-lite",
135
  contents=[prompt] + images,
136
  config=GenerateContentConfig(
137
  temperature=0.2,
 
142
  # Perform valid list generation
143
  prompt_2 = "Dari analisa diatas, Rangkum menjadi list/daftar dokumen yang valid saja. Misal: 'paspor', 'pasfoto', 'kartukeluarga', 'buktinikah', 'KTP', 'suratkelahiran', 'suratsponsor', 'suratkerja', 'NIB', 'SIUP', 'suratjaminanstaff', 'suratsekolah', 'kontrakkerja', 'suratpensiun', 'rekeningkoran', 'slipgaji', 'other'. Jangan tambahkan kata-kata atau kalimat lain."
144
  response_2 = client.models.generate_content(
145
+ model="gemini-2.0-flash-lite",
146
  contents=[analysis, prompt_2], # Pass prompt and image files
147
  config=GenerateContentConfig(
148
  temperature=0.2,
 
186
  renamed_images = []
187
  for image in images:
188
  response = client.models.generate_content(
189
+ model="gemini-2.0-flash-lite",
190
  contents=[prompt_3, image],
191
  config=GenerateContentConfig(
192
  temperature=0.2,