sandz7 commited on
Commit
b6eee7b
β€’
1 Parent(s): 91a3e4d

placed the yield for text in streamer

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
13
  DESCRIPTION = '''
14
  <div>
15
  <h1 style="text-align: center;">Loki πŸ‘οΈ</h1>
16
- <p>This uses Llama 3 and GPT-4o as generation, both of these make the final generation. <a href="https://huggingface.co/meta-llama/Meta-Llama-3-8B"><b>Llama3-8b</b></a>and <a href="https://platform.openai.com/docs/models/gpt-4o"><b>GPT-4o</b></a></p>
17
  </div>
18
  '''
19
 
@@ -68,7 +68,7 @@ def chat_llama3_8b(message: str,
68
  outputs = []
69
  for text in streamer:
70
  outputs.append(text)
71
- return outputs
72
 
73
 
74
  # Gradio block
 
13
  DESCRIPTION = '''
14
  <div>
15
  <h1 style="text-align: center;">Loki πŸ‘οΈ</h1>
16
+ <p>This uses Llama 3 and GPT-4o as generation, both of these make the final generation. <a href="https://huggingface.co/meta-llama/Meta-Llama-3-8B"><b>Llama3-8b</b></a> and <a href="https://platform.openai.com/docs/models/gpt-4o"><b>GPT-4o</b></a></p>
17
  </div>
18
  '''
19
 
 
68
  outputs = []
69
  for text in streamer:
70
  outputs.append(text)
71
+ yield "".join(outputs)
72
 
73
 
74
  # Gradio block