Spaces:

unlisboa
/

bart_qa_assistant

Runtime error

MrVicente commited on May 24, 2022

Commit

7c59cdd

1 Parent(s): a7c96e0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ from transformers import (
       BartForConditionalGeneration,
       BartTokenizer
 )
 def get_device():
     # If there's a GPU available...

       BartForConditionalGeneration,
       BartTokenizer
 )
+import torch
 def get_device():
     # If there's a GPU available...