Update src/streamlit_app.py

src/streamlit_app.py

@@ -12,7 +12,7 @@ st.set_page_config(layout="wide", page_title="Synthetic Image Generator - Condit
 
 TEXTS = {
     "en": {
-        "title": "RAYgan Datasets Generator RAY AUTRA TECHNOLOGY",
+        "title": "RAYgan Datasets Generator",
         "description": "Uses a **Conditional Generative Adversarial Network (GAN)** to generate synthetic clothing images.",
         "model_source": "Model source: [RAYAuser/raygan-zalando-datasetsgen](https://huggingface.co/RAYAuser/raygan-zalando-datasetsgen)",
         "sidebar_header": "Generation Options",
@@ -31,14 +31,14 @@ TEXTS = {
         "preview_caption": "Preview {idx}",
         "download_button": "Download ZIP file",
         "generation_success": "Generation complete and images ready for download!",
-        "model_not_found_error": "Error: The model file could not be found locally. Please ensure 'Raygan-zalando_datasetsgen.pth' is in the 'src' directory.",
+        "model_not_found_error": "Error: The model file could not be found locally. Please ensure 'Raygan-zalando_datasetsgen.pth' is in the same directory as this script.",
         "instructions_header": "Instructions:",
         "instructions_1": "1. Choose your generation mode.",
         "instructions_2": "2. Enter the number of images you want to create.",
         "instructions_3": "3. Click on 'Launch Generation'.",
     },
     "fr": {
-        "title": "RAYgan Datasets Generator RAY AUTRA TECHNOLOGY",
+        "title": "RAYgan Datasets Generator",
         "description": "Utilise un modèle **Conditional Generative Adversarial Network (GAN)** pour générer des images synthétiques de vêtements.",
         "model_source": "Source du modèle : [RAYAuser/raygan-zalando-datasetsgen](https://huggingface.co/RAYAuser/raygan-zalando-datasetsgen)",
         "sidebar_header": "Options de Génération",
@@ -57,7 +57,7 @@ TEXTS = {
         "preview_caption": "Aperçu {idx}",
         "download_button": "Télécharger le fichier ZIP",
         "generation_success": "Génération terminée et images prêtes pour le téléchargement !",
-        "model_not_found_error": "Erreur : Le fichier du modèle n'a pas pu être trouvé localement. Assurez-vous que 'Raygan-zalando_datasetsgen.pth' se trouve dans le répertoire 'src'.",
+        "model_not_found_error": "Erreur : Le fichier du modèle n'a pas pu être trouvé localement. Assurez-vous que 'Raygan-zalando_datasetsgen.pth' se trouve dans le même répertoire que ce script.",
         "instructions_header": "Instructions :",
         "instructions_1": "1. Choisissez votre mode de génération.",
         "instructions_2": "2. Entrez le nombre d'images que vous souhaitez créer.",
@@ -96,7 +96,7 @@ Z_DIM = 100
 NGF = 64
 NUM_CLASSES = 10
 IMAGE_SIZE = 32
-MODEL_FILE = "./Raygan-zalando_datasetsgen.pth"
+MODEL_FILE = "Raygan-zalando_datasetsgen.pth"
 
 class_names = [
     "T-shirt/top", "Trouser", "Pullover", "Dress", "Coat",