Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 from PIL import Image
 
 # Load the model
-model = AutoModelForImageSegmentation.from_pretrained("path/to/gelan-c-seg.pt")
+model = AutoModelForImageSegmentation.from_pretrained("./gelan-c-seg.pt")
 
 # Load the tokenizer (if needed)
 tokenizer = AutoTokenizer.from_pretrained("path/to/tokenizer")
@@ -16,7 +16,7 @@ def predict_segmentation(image):
     # Perform inference
     output = model(image)
     # Process the output as needed (e.g., post-processing for segmentation masks)
-    # (Replace this with your actual processing code)
+    
     segmentation_mask = output.logits.argmax(dim=1).squeeze().detach().numpy()
     return segmentation_mask