Hugging Face's logo

Spaces:
rairo
/
Athena-API
Running

App Files Community
rairo commited on
Commit
32e42ad
·
verified ·
1 Parent(s): a975674

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -69,7 +69,7 @@ except Exception as e:
69
  exit(1)
70
 
71
  # --- Model Constants (as per Streamlit app) ---
72
- CATEGORY_MODEL = "gemini-2.0-flash-exp"
73
  GENERATION_MODEL = "gemini-2.0-flash-exp-image-generation"
74
  #TTS_MODEL = "gemini-2.5-flash-preview-tts"
75
 
 
69
  exit(1)
70
 
71
  # --- Model Constants (as per Streamlit app) ---
72
+ CATEGORY_MODEL = "gemini-2.5-flash"
73
  GENERATION_MODEL = "gemini-2.0-flash-exp-image-generation"
74
  #TTS_MODEL = "gemini-2.5-flash-preview-tts"
75