anirudh-sub commited on
Commit
88570c9
·
1 Parent(s): 95d1712

Debugged app.py and requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +3 -1
app.py CHANGED
@@ -20,6 +20,7 @@ import transformers
20
  from transformers import AutoTokenizer
21
  import torch
22
  import gradio as gr
 
23
 
24
  model = "anirudh-sub/debate_model_v2.1"
25
  tokenizer = AutoTokenizer.from_pretrained(model)
 
20
  from transformers import AutoTokenizer
21
  import torch
22
  import gradio as gr
23
+ from diffusers.utils.torch_utils import randn_tensor
24
 
25
  model = "anirudh-sub/debate_model_v2.1"
26
  tokenizer = AutoTokenizer.from_pretrained(model)
requirements.txt CHANGED
@@ -1,3 +1,5 @@
1
  transformers
2
  torch
3
- gradio
 
 
 
1
  transformers
2
  torch
3
+ gradio
4
+ diffusers.utils.torch_utils
5
+ huggingface-hub