PawelMar commited on
Commit
c526faa
·
verified ·
1 Parent(s): 4e05e43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModel
5
  def greet(name):
6
  return "Hello " + name + "!!"
7
 
8
- model = AutoModel.from_pretrained("models/last.pt")
9
  pipe = pipeline("image-segmentation", model = model)
10
  iface = gr.Interface.from_pipeline(pipe)
11
  # iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
5
  def greet(name):
6
  return "Hello " + name + "!!"
7
 
8
+ model = AutoModel.from_pretrained("./models/")
9
  pipe = pipeline("image-segmentation", model = model)
10
  iface = gr.Interface.from_pipeline(pipe)
11
  # iface = gr.Interface(fn=greet, inputs="text", outputs="text")