fffiloni commited on
Commit
c93d85e
1 Parent(s): 396214f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,7 +70,7 @@ def get_magnet(prompt):
70
  print(amended_prompt)
71
  client = Client("https://fffiloni-magnet.hf.space/")
72
  result = client.predict(
73
- "facebook/magnet-small-10secs", # Literal['facebook/magnet-small-10secs', 'facebook/magnet-medium-10secs', 'facebook/magnet-small-30secs', 'facebook/magnet-medium-30secs', 'facebook/audio-magnet-small', 'facebook/audio-magnet-medium'] in 'Model' Radio component
74
  "", # str in 'Model Path (custom models)' Textbox component
75
  amended_prompt, # str in 'Input Text' Textbox component
76
  3, # float in 'Temperature' Number component
@@ -126,7 +126,7 @@ def infer(image_in):
126
 
127
  music_o = get_magnet(cleaned_text)
128
 
129
- return cleaned_text, music_o
130
 
131
  title = "Image to Music V2",
132
  description = "Get music from a picture"
 
70
  print(amended_prompt)
71
  client = Client("https://fffiloni-magnet.hf.space/")
72
  result = client.predict(
73
+ "facebook/magnet-medium-10secs", # Literal['facebook/magnet-small-10secs', 'facebook/magnet-medium-10secs', 'facebook/magnet-small-30secs', 'facebook/magnet-medium-30secs', 'facebook/audio-magnet-small', 'facebook/audio-magnet-medium'] in 'Model' Radio component
74
  "", # str in 'Model Path (custom models)' Textbox component
75
  amended_prompt, # str in 'Input Text' Textbox component
76
  3, # float in 'Temperature' Number component
 
126
 
127
  music_o = get_magnet(cleaned_text)
128
 
129
+ return cleaned_text.lstrip("\n"), music_o
130
 
131
  title = "Image to Music V2",
132
  description = "Get music from a picture"