Spaces:

JoannaKOKO
/

Tarot_Explainer_gpu

Sleeping

JoannaKOKO commited on Mar 25

Commit

9e47dd3

verified ·

1 Parent(s): 7c28d34

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ Provide a professional interpretation covering:
     generated_ids = [output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)]
     # Move output to CPU before decoding
-    response = tokenizer.batch_decode(generated_ids.cpu(), skip_special_tokens=True)[0]
     return response
 def main():

     generated_ids = [output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)]
     # Move output to CPU before decoding
+    response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
     return response
 def main():