anirudh-sub commited on
Commit
a83fa2d
·
1 Parent(s): de2fd00

Added an import

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import transformers
3
  from transformers import AutoTokenizer
 
4
 
5
  model = "anirudh-sub/debate_model_v2.1"
6
  tokenizer = AutoTokenizer.from_pretrained(model)
 
1
  import gradio as gr
2
  import transformers
3
  from transformers import AutoTokenizer
4
+ import torch
5
 
6
  model = "anirudh-sub/debate_model_v2.1"
7
  tokenizer = AutoTokenizer.from_pretrained(model)