Update app.py

app.py

@@ -94,7 +94,7 @@ from transformers import pipeline
 zephyr_model = "HuggingFaceH4/zephyr-7b-beta"
 mixtral_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
-pipe = pipeline("text-generation", model=mixtral_model, torch_dtype=torch.bfloat16, device_map="auto")
+pipe = pipeline("text-generation", model=zephyr_model, torch_dtype=torch.bfloat16, device_map="auto")
 
 agent_maker_sys = f"""
 You are an AI whose job is to help users create their own music which its genre will reflect the character or scene from an image described by users.
@@ -118,7 +118,7 @@ def infer(image_in):
     prompt = f"{instruction.strip()}\n{user_prompt}</s>"    
     #print(f"PROMPT: {prompt}")
     
-    gr.Info("Building a system according to the image caption ...")
+    gr.Info("Building a musical prompt according to the image caption ...")
     outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
     
 
@@ -127,11 +127,12 @@ def infer(image_in):
     
     print(f"SUGGESTED Musical prompt: {cleaned_text}")
 
+    gr.Info("Now calling MAGNet for music ...")
     music_o = get_magnet(cleaned_text)
     
     return cleaned_text.lstrip("\n"), music_o
 
-title = "Image to Music V2",
+demo_title = "Image to Music V2"
 description = "Get music from a picture"
 
 css = """
@@ -145,7 +146,7 @@ css = """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.HTML(f"""
-        <h2 style="text-align: center;">{title}</h2>
+        <h2 style="text-align: center;">{demo_title}</h2>
         <p style="text-align: center;">{description}</p>
         """)
         
@@ -156,7 +157,7 @@ with gr.Blocks(css=css) as demo:
                     type = "filepath",
                     elem_id = "image-in"
                 )
-                submit_btn = gr.Button("Make LLM system from my pic !")
+                submit_btn = gr.Button("Make music from my pic !")
             with gr.Column():
                 caption = gr.Textbox(
                     label = "Musical prompt"