img-to-music / app.py
fffiloni's picture
Update app.py
1c0a21f
import gradio as gr
import os
img_to_text = gr.Blocks.load(name="spaces/pharma/CLIP-Interrogator")
text_to_music = gr.Interface.load("spaces/Mubert/Text-to-Music")
def get_prompts(uploaded_image):
print("calling Clip interrogator ...")
prompt = img_to_text(uploaded_image, fn_index=1)[0]
print(f"""β€”β€”β€”
Got prompt result:
{prompt}
β€”β€”β€”β€”β€”β€”β€”
""")
music_result = get_music(prompt)
return music_result
def get_music(prompt):
print("calling now mubert ....")
email = "blabla@mail.com"
duration = 30
#result = text_to_music(email, prompt, duration)
#output = os.path.join(result)
#print(output)
return prompt
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
input_img = gr.Image(type="filepath")
generate = gr.Button("Generate Music from Image")
with gr.Column():
music_output = gr.Textbox(label="Result")
generate.click(get_prompts, inputs=[input_img], outputs=[music_output])
demo.launch()