Spaces:

bkoz
/

bk-sandbox

Running on Zero

bkoz commited on May 18

Commit

866f150

•

1 Parent(s): ec23033

testing vscode

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_
 def generate(prompt):
     response = pipe(prompt, max_new_tokens=512)
     # r = response[0]['generated_text'][-1]['content']
-    print(f'response!')
     r = response[0]['generated_text']
     return r

 def generate(prompt):
     response = pipe(prompt, max_new_tokens=512)
     # r = response[0]['generated_text'][-1]['content']
+    print(f'Response received!')
     r = response[0]['generated_text']
     return r