pengpeng2000 commited on
Commit
e50e582
·
verified ·
1 Parent(s): 1088779

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,5 +1,10 @@
1
  import gradio as gr
2
 
 
 
 
 
 
3
  def greet(name):
4
  return "Hello " + name + "!!"
5
 
 
1
  import gradio as gr
2
 
3
+ from transformers import AutoProcessor, AutoModelForMaskGeneration
4
+
5
+ processor = AutoProcessor.from_pretrained("facebook/sam-vit-base")
6
+ model = AutoModelForMaskGeneration.from_pretrained("facebook/sam-vit-base")
7
+
8
  def greet(name):
9
  return "Hello " + name + "!!"
10