yhj137 commited on
Commit
73e4a98
·
1 Parent(s): 5a00915
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
-
3
  def greet(name):
4
  return "Hello " + name + "!!"
5
 
6
  from transformers import AutoModel, AutoTokenizer
7
 
8
- model = AutoModel.from_pretrained("yhj137/pianist-transformer-rendering")
9
  demo = gr.Interface(fn=greet, inputs="text", outputs="text")
10
  demo.launch()
 
1
  import gradio as gr
2
+ import os
3
  def greet(name):
4
  return "Hello " + name + "!!"
5
 
6
  from transformers import AutoModel, AutoTokenizer
7
 
8
+ model = AutoModel.from_pretrained("yhj137/pianist-transformer-rendering", token=os.environ["hf_token"])
9
  demo = gr.Interface(fn=greet, inputs="text", outputs="text")
10
  demo.launch()