highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-11-18 20:26:34 +01:00
Code Issues Packages Projects Releases Wiki Activity

unify venv calling, add basic huggingchat module

Browse Source
This commit is contained in:
Believethehype
2023-12-29 00:27:34 +01:00
parent 2c8e81d34a
commit 6de23645e4
23 changed files with 168 additions and 200 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
nostr_dvm/tasks/imagegeneration_openai_dalle.py
View File

@@ -6,7 +6,7 @@ from io import BytesIO
import requests
from PIL import Image
from nostr_dvm.interfaces.dvmtaskinterface import DVMTaskInterface
from nostr_dvm.interfaces.dvmtaskinterface import DVMTaskInterface, process_venv
from nostr_dvm.utils.admin_utils import AdminConfig
from nostr_dvm.utils.definitions import EventDefinitions
from nostr_dvm.utils.dvmconfig import DVMConfig, build_default_config
@@ -152,17 +152,5 @@ def build_example(name, identifier, admin_config):
return ImageGenerationDALLE(name=name, dvm_config=dvm_config, nip89config=nip89config, admin_config=admin_config)
def process_venv():
args = DVMTaskInterface.process_args()
dvm_config = build_default_config(args.identifier)
dvm = ImageGenerationDALLE(name="", dvm_config=dvm_config, nip89config=NIP89Config(), admin_config=None)
result = ""
while result == "":
result = dvm.process(json.loads(args.request))
time.sleep(10)
DVMTaskInterface.write_output(result, args.output)
if __name__ == '__main__':
process_venv()
process_venv(ImageGenerationDALLE)