Variables for model names

app.py

@@ -1,3 +1,6 @@
+model_repo_id = "Freepik/F-Lite"
+model_name = "F Lite" 
+
 from dotenv import load_dotenv
 import gradio as gr
 import numpy as np
@@ -28,7 +31,6 @@ else:
     logging.warning("GEMINI_API_KEY not found in environment variables. Prompt enrichment will not work.")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "Freepik/F-Lite"
 
 if torch.cuda.is_available():
     torch_dtype = torch.bfloat16
@@ -200,7 +202,7 @@ css = """
 
 with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(" # F-Lite Text-to-Image Demo")
+        gr.Markdown(f" # {model_name} Text-to-Image Demo")
 
         with gr.Row(elem_classes="prompt-row"):
             prompt = gr.Text(
@@ -322,6 +324,9 @@ with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
             example_labels=[ex[0][:max_length] + "..." if len(ex[0]) > max_length else ex[0] for ex in examples]
         )
     
+        # Add link to model card
+        gr.Markdown(f"[{model_name} Model Card and Weights](https://huggingface.co/{model_repo_id})")
+
     # Update width and height when resolution dropdown changes
     resolution_dropdown.change(
         fn=update_resolution,
@@ -347,8 +352,6 @@ with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
         outputs=[result, seed, enriched_prompt_display, enriched_prompt_text, enrichment_error],
     )
     
-    # Add link to model card
-    gr.Markdown("[F-Lite Model Card and Weights](https://huggingface.co/Freepik/F-Lite)")
 
 if __name__ == "__main__":
     demo.launch() # server_name="0.0.0.0", share=True)