Spaces:

xTorch8
/

bart-summarizer

Sleeping

xTorch8 commited on May 21

Commit

e417e31

1 Parent(s): 1096299

Import torch

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 MODEL = "xTorch8/fine-tuned-bart"
 TOKEN = os.getenv("TOKEN")

 import gradio as gr
 import os
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+import torch
 MODEL = "xTorch8/fine-tuned-bart"
 TOKEN = os.getenv("TOKEN")