John6666 commited on
Commit
f1a369e
β€’
1 Parent(s): ed62071

Update app.py

Browse files

HF spec change.

Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,8 +1,6 @@
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
- import spaces
4
-
5
-
6
 
7
  model = AutoModelForCausalLM.from_pretrained("instruction-pretrain/instruction-synthesizer")
8
  model.to('cuda')
 
1
+ import spaces
2
  import gradio as gr
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 
4
 
5
  model = AutoModelForCausalLM.from_pretrained("instruction-pretrain/instruction-synthesizer")
6
  model.to('cuda')