From 0508489d14f043eede93bea2f37eb4d9602cf97d Mon Sep 17 00:00:00 2001
From: GF <149839341+gorillaframeai@users.noreply.github.com>
Date: Sun, 2 Feb 2025 19:14:44 +0300
Subject: [PATCH] Update gfrbmg2.py

---
 gfrbmg2.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/gfrbmg2.py b/gfrbmg2.py
index 46dee15..93a0da7 100644
--- a/gfrbmg2.py
+++ b/gfrbmg2.py
@@ -1,5 +1,8 @@
-import torch, os
-import torch.nn.functional as F
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+import torch
 import folder_paths
 from PIL import Image
 from transformers import AutoModelForImageSegmentation
@@ -7,7 +10,6 @@ from torchvision import transforms
 from torchvision.transforms.functional import normalize
 import numpy as np
 import cv2
-import sys
 import subprocess
 import BEN2