Docstring

app.py

@@ -90,6 +90,32 @@ def format_conversation_history(chat_history):
 
 @spaces.GPU(duration=120)
 def generate_response(input_data, chat_history, max_new_tokens, system_prompt, temperature, top_p, top_k, repetition_penalty):
+    """
+    Creates silly song lyrics in Hebrew based on user input and conversation history.
+
+    Args:
+        input_data (dict or str): 
+            - If dict: must include 'text' (str) and optional 'files' (list of image/video file paths).
+            - If str: treated as plain text input.
+        chat_history (list of dict): 
+            Sequence of past messages, each with keys 'role' and 'content'.
+        max_new_tokens (int): 
+            Maximum number of tokens to generate for the response.
+        system_prompt (str): 
+            Optional system-level instruction to guide the style and content of the response.
+        temperature (float): 
+            Sampling temperature; higher values yield more diverse outputs.
+        top_p (float): 
+            Nucleus sampling threshold for cumulative probability selection.
+        top_k (int): 
+            Limits sampling to the top_k most likely tokens at each step.
+        repetition_penalty (float): 
+            Penalty factor to discourage the model from repeating the same tokens.
+
+    Yields:
+        str: Streaming chunks of the generated Hebrew song lyrics in real time.
+    """
+    
     if isinstance(input_data, dict) and "text" in input_data:
         text = input_data["text"]
         files = input_data.get("files", [])