Spaces:

Parthebhan
/

segmentation

Sleeping

Parthebhan commited on Apr 22, 2024

Commit

829b850

verified ·

1 Parent(s): ae3d5be

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 from PIL import Image
 # Load the model
-model = AutoModelForImageSegmentation.from_pretrained("./gelan-c-seg.pt")
 # Load the tokenizer (if needed)
 tokenizer = AutoTokenizer.from_pretrained("path/to/tokenizer")

 from PIL import Image
 # Load the model
+model = AutoModelForImageSegmentation.from_pretrained("/absolute/path/to/gelan-c-seg.pt")")
 # Load the tokenizer (if needed)
 tokenizer = AutoTokenizer.from_pretrained("path/to/tokenizer")