mirror of
https://github.com/believethehype/nostrdvm.git
synced 2025-10-11 05:43:22 +02:00
added huggingchat examples
This commit is contained in:
14
main.py
14
main.py
@@ -7,7 +7,7 @@ from nostr_dvm.bot import Bot
|
||||
from nostr_dvm.tasks import videogeneration_replicate_svd, imagegeneration_replicate_sdxl, textgeneration_llmlite, \
|
||||
trending_notes_nostrband, discovery_inactive_follows, translation_google, textextraction_pdf, \
|
||||
translation_libretranslate, textextraction_google, convert_media, imagegeneration_openai_dalle, texttospeech, \
|
||||
imagegeneration_sd21_mlx, advanced_search, textgeneration_huggingchat
|
||||
imagegeneration_sd21_mlx, advanced_search, textgeneration_huggingchat, summarization_huggingchat
|
||||
from nostr_dvm.utils.admin_utils import AdminConfig
|
||||
from nostr_dvm.utils.backend_utils import keep_alive
|
||||
from nostr_dvm.utils.definitions import EventDefinitions
|
||||
@@ -155,10 +155,16 @@ def playground():
|
||||
bot_config.SUPPORTED_DVMS.append(mlx)
|
||||
mlx.run()
|
||||
|
||||
if os.getenv("HUGGINGFACE_EMAIL") is not None and os.getenv("HUGGINGFACE_EMAIL") != "":
|
||||
hugginchat = textgeneration_huggingchat.build_example("Huggingchat", "huggingchat",admin_config)
|
||||
bot_config.SUPPORTED_DVMS.append(hugginchat)
|
||||
hugginchat.run()
|
||||
|
||||
hugginchatsum = summarization_huggingchat.build_example("Huggingchat Summarizer", "huggingchatsum", admin_config)
|
||||
bot_config.SUPPORTED_DVMS.append(hugginchatsum)
|
||||
hugginchatsum.run()
|
||||
|
||||
|
||||
hugginchat = textgeneration_huggingchat.build_example("Huggingchat", "hugginchat", admin_config)
|
||||
bot_config.SUPPORTED_DVMS.append(hugginchat)
|
||||
hugginchat.run()
|
||||
|
||||
# Run the bot
|
||||
Bot(bot_config)
|
||||
|
Reference in New Issue
Block a user