fffiloni commited on
Commit
f372cec
·
1 Parent(s): 9bcd336

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import time
2
-
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
 
@@ -36,17 +36,19 @@ def get_track_by_tags(tags, pat, duration, maxit=20, loop=False):
36
  for i in range(maxit):
37
  r = httpx.get(trackurl)
38
  if r.status_code == 200:
39
- return trackurl
 
 
40
  time.sleep(1)
41
 
42
 
43
  def generate_track_by_prompt(prompt):
44
- #try:
45
- # pat = get_pat("[email protected]")
46
- # _, tags = get_tags_for_prompts(minilm, mubert_tags_embeddings, [prompt, ])[0]
47
- # return get_track_by_tags(tags, pat, int(30), loop=False)
48
- #except Exception as e:
49
- return "hello it is me"
50
 
51
 
52
  iface = gr.Interface(fn=generate_track_by_prompt, inputs=["text"], outputs=[gr.Text(label="Result")])
 
1
  import time
2
+ import base64
3
  import gradio as gr
4
  from sentence_transformers import SentenceTransformer
5
 
 
36
  for i in range(maxit):
37
  r = httpx.get(trackurl)
38
  if r.status_code == 200:
39
+ data = open(trackurl, "r").read()
40
+ encoded = base64.b64encode(data)
41
+ return encoded
42
  time.sleep(1)
43
 
44
 
45
  def generate_track_by_prompt(prompt):
46
+ try:
47
+ pat = get_pat("[email protected]")
48
+ _, tags = get_tags_for_prompts(minilm, mubert_tags_embeddings, [prompt, ])[0]
49
+ return get_track_by_tags(tags, pat, int(30), loop=False)
50
+ except Exception as e:
51
+ return str(e)
52
 
53
 
54
  iface = gr.Interface(fn=generate_track_by_prompt, inputs=["text"], outputs=[gr.Text(label="Result")])