Update app.py

app.py

@@ -15,6 +15,8 @@ def query(payload):
 
 def generate_and_append_text(max_length):
     global generated_text
+    # Ensure that the input is not empty
+    input_text = generated_text if generated_text.strip() else " "
     parameters = {
         "max_new_tokens": max_length,
         "top_p": 0.9,
@@ -24,14 +26,15 @@ def generate_and_append_text(max_length):
         "length_penalty": 0.0,
         "eos_token_id": None,
     }
-    payload = {"inputs": generated_text, "parameters": parameters, "options": {"use_cache": False}}
+    payload = {"inputs": input_text, "parameters": parameters, "options": {"use_cache": False}}
     data = query(payload)
     if "error" in data:
         return f"<span style='color:red'>ERROR: {data['error']} </span>"
-    new_text = data[0]["generated_text"].replace(generated_text, "").strip()
-    generated_text += " " + new_text
+    new_text = data[0]["generated_text"].replace(input_text, "").strip()
+    generated_text += " " + new_text if generated_text.strip() else new_text
     return generated_text
 
+
 if __name__ == "__main__":
     demo = gr.Blocks()
     with demo: