bkoz commited on
Commit
aa41904
·
1 Parent(s): 9030a60
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_
16
  @spaces.GPU
17
  def generate(prompt):
18
  response = pipe(prompt, max_new_tokens=512)
19
- r = response[0]['generated_text'][-1]['content']
20
- return r
21
 
22
  gr.Interface(
23
  fn=generate,
 
16
  @spaces.GPU
17
  def generate(prompt):
18
  response = pipe(prompt, max_new_tokens=512)
19
+ # r = response[0]['generated_text'][-1]['content']
20
+ return response
21
 
22
  gr.Interface(
23
  fn=generate,