11

app.py

@@ -30,6 +30,7 @@ def convert_prompt(
     if not os.environ.get("OPENAI_API_KEY"):
         return prompt
     client = OpenAI()
+    prompt = clean_string(prompt)
     for i in range(retry_times):
         try:
             response = client.chat.completions.create(
@@ -72,7 +73,7 @@ def convert_prompt(
                     },
                     {
                         "role": "user",
-                        "content": "Create an imaginative image descriptive caption for the user input : \"$PROMPT\""
+                        "content": f"Create an imaginative image descriptive caption for the user input : {prompt}"
                     }
                 ],
                 model="glm-4-plus",
@@ -136,8 +137,8 @@ examples = [
 
 
 with gr.Blocks() as demo:
-    with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""
+    with gr.Column():
+        gr.Markdown("""
             <div style="text-align: center; font-size: 32px; font-weight: bold; margin-bottom: 20px;">
              CogView4-6B Huggingface Space🤗
            </div>