fffiloni commited on
Commit
0b10fd7
1 Parent(s): caf7b05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -71,20 +71,20 @@ def infer(audio_file):
71
 
72
  result = client.predict(
73
  llama_q, # str in 'Message' Textbox component
74
- api_name="/chat"
75
  )
76
 
77
 
78
- final_res = result.outputs()[-1]
79
 
80
- print(final_res)
 
81
 
82
- images = pipe(prompt=final_res).images[0]
83
 
84
  print("Finished")
85
 
86
  #return cap_result, result, images
87
- return images, final_res, gr.update(visible=True)
88
 
89
  css = """
90
  #col-container {max-width: 510px; margin-left: auto; margin-right: auto;}
 
71
 
72
  result = client.predict(
73
  llama_q, # str in 'Message' Textbox component
74
+ api_name="/chat_1"
75
  )
76
 
77
 
 
78
 
79
+
80
+ print(f"Llama2 result: {result}")
81
 
82
+ images = pipe(prompt=result).images[0]
83
 
84
  print("Finished")
85
 
86
  #return cap_result, result, images
87
+ return images, result, gr.update(visible=True)
88
 
89
  css = """
90
  #col-container {max-width: 510px; margin-left: auto; margin-right: auto;}