fffiloni commited on
Commit
88cbac0
·
verified ·
1 Parent(s): 6e65224

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -4,7 +4,7 @@ import json
4
  import re
5
  import random
6
  import numpy as np
7
- from gradio_client import Client
8
 
9
  MAX_SEED = np.iinfo(np.int32).max
10
 
@@ -48,14 +48,15 @@ def get_outfit_prompt(user_prompt):
48
  def get_salmonn(audio_in, prompt):
49
  client = Client("fffiloni/SALMONN-7B-gradio")
50
  result = client.predict(
51
- audio_in, # filepath in 'Audio' Audio component
52
- prompt, # str in 'User question' Textbox component
53
- 4, # float (numeric value between 1 and 10) in 'beam search numbers' Slider component
54
- 1, # float (numeric value between 0.8 and 2.0) in 'temperature' Slider component
55
- 0.9, # float (numeric value between 0.1 and 1.0) in 'top p' Slider component
56
- api_name="/gradio_answer"
57
  )
58
  print(result)
 
59
  return result
60
 
61
  def infer(audio_in):
 
4
  import re
5
  import random
6
  import numpy as np
7
+ from gradio_client import Client, handle_file
8
 
9
  MAX_SEED = np.iinfo(np.int32).max
10
 
 
48
  def get_salmonn(audio_in, prompt):
49
  client = Client("fffiloni/SALMONN-7B-gradio")
50
  result = client.predict(
51
+ speech=handle_file(audio_in),
52
+ text_input=prompt,
53
+ num_beams=4,
54
+ temperature=1,
55
+ top_p=0.9,
56
+ api_name="/gradio_answer_1"
57
  )
58
  print(result)
59
+
60
  return result
61
 
62
  def infer(audio_in):