Merge branch 'main' of hf.co:spaces/PawelMar/LettuceSegmentAPI into add_model

app.py

@@ -1,7 +1,12 @@
 import gradio as gr
+from transformers import AutoModel
+
 
 def greet(name):
     return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+model = AutoModel.from_pretrained("models/last.pt")
+pipe = pipeline("image-segmentation", model = model)
+iface = gr.Interface.from_pipeline(pipe)
+# iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()