Fixed folder errors

Author: uesleibros

opengpt/__init__.py

@@ -20,17 +20,14 @@ def Print(context: Text, warn: Optional[bool] = False) -> None:
 class OpenGPT:
 	@classmethod
 	def __init__(self: type, provider: Text, type: Optional[Text] = "completion", options: Optional[Union[Dict[Text, Text], None]] = None) -> None:
-	
-		self.__DIR: Text = os.getcwd()	
+		self.__DIR: Text = os.path.dirname(os.path.abspath(__file__))
 		self.__LoadModels()
 		self.__TYPE: Text = type
 		self.__OPTIONS: Union[Dict[Text, Text], None] = options
 		self.__PROVIDER: Text = provider
-
 		self.__Verifications()
 		self.__MODULE: module = importlib.import_module(f"opengpt.models.{self.__TYPE}.{self.__PROVIDER}.model")
 		self.__MODEL_CLASS: type = getattr(self.__MODULE, "Model")
-
 		self.__model: type = None
 		self.__InitializeModelClass()
 
@@ -58,7 +55,7 @@ def __InitializeModelClass(self: type) -> None:
 
 	@classmethod
 	def __LoadModels(self: type) -> None:
-		self.__DATA: Dict[Text, Text] = yaml.safe_load(open(self.__DIR + "/config.yml").read())
+		self.__DATA: Dict[Text, Text] = yaml.safe_load(open(self.__DIR + "/config.yml", "r").read())
 
 	@classmethod
 	def __Verifications(self: type) -> None:

opengpt/models/image/hotpot/model.py

@@ -23,8 +23,8 @@ class Model:
 	@classmethod
 	def __init__(self: type, style: Optional[Text] = "Hotpot Art 9") -> None:
 		self._SETUP_LOGGER()
+		self.__DIR: Text = os.path.dirname(os.path.abspath(__file__))
 		self.__LoadStyles()
-		self.__DIR: Text = os.getcwd()
 		self.__session: requests.Session = requests.Session()
 		self.__UNIQUE_ID: str = UniqueID(16)
 		self.STYLE: Text = style
@@ -73,7 +73,7 @@ def UpdateStyle(self: type, style: Text) -> None:
 
 	@classmethod
 	def __LoadStyles(self: type) -> None:
-		self.__DATA: Dict[Text, Text] = yaml.safe_load(open(self.__DIR + "/styles.yml").read())
+		self.__DATA: Dict[Text, Text] = yaml.safe_load(open(self.__DIR + "/styles.yml", "r").read())
 
 	@classmethod
 	def __Fields(self: type, *args: Tuple[int, str], **kwargs: Dict[str, Any]) -> Text:

tests/test.py

@@ -1,4 +1,4 @@
 from opengpt import OpenGPT
 
-hotpot = OpenGPT(provider="hotpot", type="image", options={"style": "Acrylic Art"})
-print(hotpot.Generate("Man with Black T-Shirt").url)
+hotpot = OpenGPT(provider="hotpot", type="image", options={"style": "Portrait Anime 1"})
+print(hotpot.Generate("Man with Red T-Shirt and Blue Light Hair").url)