Update app.py

app.py

@@ -65,7 +65,7 @@ from transformers import pipeline
 pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
 
 @spaces.GPU(enable_queue=True)
-def get_llm_idea(user_prompt):
+def get_card_idea(user_prompt):
     agent_maker_sys = f'''
 Your job is to generate new magic card from an image description. 
 
@@ -113,16 +113,16 @@ def infer(image_in, cap_type):
     elif cap_type == "Literal" :
         user_prompt = get_caption_from_kosmos(image_in)
     
-    gr.Info("Building a system according to the image caption ...")
-    outputs = get_llm_idea(user_prompt)
+    gr.Info("Building a new card according to the image caption ...")
+    outputs = get_card_idea(user_prompt)
     
 
     pattern = r'\<\|system\|\>(.*?)\<\|assistant\|\>'
     cleaned_text = re.sub(pattern, '', outputs[0]["generated_text"], flags=re.DOTALL)
     
-    print(f"SUGGESTED LLM: {cleaned_text}")
+    print(f"SUGGESTED CARD: {cleaned_text}")
     
-    return user_prompt, cleaned_text.lstrip("\n")
+    return cleaned_text.lstrip("\n")
 
 title = f"Magic Card Generator",
 description = f""
@@ -142,7 +142,7 @@ div#component-14 > .gallery > .gallery-item > .container > img {
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.HTML(f"""
-        <h2 style="text-align: center;">LLM Agent from a Picture</h2>
+        <h2 style="text-align: center;">Magic Card Generator</h2>
         <p style="text-align: center;">{description}</p>
         """)
         
@@ -182,7 +182,6 @@ with gr.Blocks(css=css) as demo:
             cap_type
         ],
         outputs =[
-            caption,
             result
         ]
     )