Chris Addis commited on
Commit ·
3830538
1
Parent(s): 2cfe564
more tests
Browse files
app.py
CHANGED
|
@@ -137,6 +137,12 @@ def create_demo():
|
|
| 137 |
##########################################################################
|
| 138 |
# Function to check authorization
|
| 139 |
def check_authorization(profile: gr.OAuthProfile | None):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 140 |
|
| 141 |
is_authorized = profile.username in AUTHORIZED_USER_IDS
|
| 142 |
if is_authorized:
|
|
@@ -164,8 +170,6 @@ def create_demo():
|
|
| 164 |
|
| 165 |
return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
|
| 166 |
else:
|
| 167 |
-
all_models_list = preferred_models
|
| 168 |
-
|
| 169 |
# Default model value
|
| 170 |
default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
|
| 171 |
text = f"""**Current Model**: Llama 4 Maverick (free)
|
|
|
|
| 137 |
##########################################################################
|
| 138 |
# Function to check authorization
|
| 139 |
def check_authorization(profile: gr.OAuthProfile | None):
|
| 140 |
+
if profile is None:
|
| 141 |
+
# Default model value
|
| 142 |
+
default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
|
| 143 |
+
text = f"""**Current Model**: Llama 4 Maverick (free)
|
| 144 |
+
**Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
|
| 145 |
+
return gr.update(choices=preferred_models, label="Select Model",value=default_model),True,text
|
| 146 |
|
| 147 |
is_authorized = profile.username in AUTHORIZED_USER_IDS
|
| 148 |
if is_authorized:
|
|
|
|
| 170 |
|
| 171 |
return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
|
| 172 |
else:
|
|
|
|
|
|
|
| 173 |
# Default model value
|
| 174 |
default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
|
| 175 |
text = f"""**Current Model**: Llama 4 Maverick (free)
|