Fixed model pull by name

This commit is contained in:
jeffser 2024-09-02 16:04:17 -06:00
parent 08c0c54b98
commit a08be7351c

View File

@ -3,7 +3,7 @@
Handles UI dialogs Handles UI dialogs
""" """
import os import os
import logging, requests import logging, requests, threading
from pytube import YouTube from pytube import YouTube
from html2text import html2text from html2text import html2text
from gi.repository import Adw, Gtk from gi.repository import Adw, Gtk
@ -278,7 +278,7 @@ def create_model_from_file(self):
def create_model_from_name_response(self, dialog, task, entry): def create_model_from_name_response(self, dialog, task, entry):
model = entry.get_text().lower().strip() model = entry.get_text().lower().strip()
if dialog.choose_finish(task) == 'accept' and model: if dialog.choose_finish(task) == 'accept' and model:
self.pull_model(model) threading.Thread(target=self.model_manager.pull_model, kwargs={"model_name": model}).start()
def create_model_from_name(self): def create_model_from_name(self):
entry = Gtk.Entry() entry = Gtk.Entry()