adds docstrings

app.py

@@ -7,7 +7,6 @@ import spaces
 default_model = "lucianosb/acaraje-brazil-xl"
 default_prompt = "delicious acarajé"
 
-@spaces.GPU
 pipeline = StableDiffusionXLPipeline.from_single_file(
             "https://huggingface.co/lucianosb/sinteticoXL-models/blob/main/sinteticoXL_v1dot2.safetensors",
             torch_dtype=torch.float16,
@@ -16,6 +15,16 @@ pipeline = StableDiffusionXLPipeline.from_single_file(
         ).to("cuda")
 
 def make_description(model):
+    """
+    Generates a description based on the given model.
+
+    Parameters:
+        model (str): The name of the model.
+
+    Returns:
+        str: The generated description.
+
+    """
     pipeline.unload_lora_weights()
     model_keywords = {
         "lucianosb/acaraje-brazil-xl": ["acarajé", "acaraje"],
@@ -36,6 +45,16 @@ def make_description(model):
     return "Triggered with the following keywords: \n\n" + "- " + "\n- ".join(keywords)
 
 def make_prompt(model):
+    """
+    Generates a prompt based on the given model.
+
+    Parameters:
+        model (str): The name of the model.
+
+    Returns:
+        str: The generated prompt.
+
+    """
     prompts = {
         "lucianosb/acaraje-brazil-xl": "delicious acarajé",
         "lucianosb/boto-brazil-xl": "beautiful boto in a river",
@@ -55,6 +74,19 @@ def make_prompt(model):
 
 @spaces.GPU
 def make_image(model, prompt):
+    """
+    Generates an image based on the given model and prompt.
+
+    Args:
+        model (str): The name of the model.
+        prompt (str): The prompt to generate the image.
+
+    Returns:
+        PIL.Image.Image: The generated image.
+
+    Raises:
+        ValueError: If the model is invalid.
+    """
     weight_file_names = {
         "lucianosb/acaraje-brazil-xl": "Acarajé_-_Brazil_XL.safetensors",
         "lucianosb/boto-brazil-xl": "Boto_-_Brazil_XL.safetensors",