chore: add missing requirements

- add missing requirements
- add dotenv file loading
This commit is contained in:
BuildTools 2024-08-17 13:27:18 -07:00
parent 03c8a23e07
commit a1650bba07
No known key found for this signature in database
GPG Key ID: 3270C066C15D530B
2 changed files with 7 additions and 2 deletions

View File

@ -1,9 +1,11 @@
psutil~=6.0.0
requests~=2.32.3
numpy<2.0.0
torch~=2.4.0
torch~=1.13.1
sentencepiece~=0.2.0
PyYAML~=6.0.2
pynvml~=11.5.3
PySide6~=6.7.2
flask~=3.0.3
flask~=3.0.3
python-dotenv~=1.0.1
safetensors~=0.4.4

View File

@ -4,6 +4,7 @@
from functools import partial
from datetime import datetime
from dotenv import load_dotenv
from PySide6.QtCore import *
from PySide6.QtGui import *
from PySide6.QtWidgets import *
@ -42,6 +43,8 @@ def __init__(self):
self.setGeometry(100, 100, width, height)
self.setWindowFlag(Qt.FramelessWindowHint)
load_dotenv() # Loads the .env file
self.resize_factor = 1.1 # 10% increase/decrease
self.default_width, self.default_height = self.parse_resolution()
self.resize(self.default_width, self.default_height)