Hugging Face's logo

Spaces:
rairo
/
Athena-API
Running

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

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -70,7 +70,8 @@ except Exception as e:
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
 
76
 
 
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
+ GENERATION_MODEL = "gemini-2.5-flash-image-preview"
75
  #TTS_MODEL = "gemini-2.5-flash-preview-tts"
76
 
77