image-to-music-v2

Running on Zero

fffiloni commited on Dec 21, 2023

Commit

09d059c

•

1 Parent(s): bf18300

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -76,7 +76,7 @@ instruction = f"""
 """
 def infer(image_in):
-    gr.Info("Getting image caption with Fuyu...")
     user_prompt = get_caption(image_in)
     prompt = f"{instruction.strip()}\n{user_prompt}</s>"
@@ -121,7 +121,7 @@ with gr.Blocks(css=css) as demo:
                 submit_btn = gr.Button("Make LLM system from my pic !")
             with gr.Column():
                 caption = gr.Textbox(
-                    label = "Image caption (fuyu)",
                     elem_id = "image-caption"
                 )
                 result = gr.Textbox(

 """
 def infer(image_in):
+    gr.Info("Getting image caption with Kosmos2...")
     user_prompt = get_caption(image_in)
     prompt = f"{instruction.strip()}\n{user_prompt}</s>"
                 submit_btn = gr.Button("Make LLM system from my pic !")
             with gr.Column():
                 caption = gr.Textbox(
+                    label = "Image caption (Kosmos2)",
                     elem_id = "image-caption"
                 )
                 result = gr.Textbox(