hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added 'Image recognition' text to available models, also updated list of models

Browse Source
This commit is contained in:
jeffser 2024-05-24 15:46:02 -06:00
parent 999f7a7151
commit 0c9fd6685b
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/available_models.py
View File

File diff suppressed because one or more lines are too long

5
src/window.py
View File

@ -212,7 +212,7 @@ class AlpacaWindow(Adw.ApplicationWindow):
def verify_if_image_can_be_used(self, pspec=None, user_data=None): def verify_if_image_can_be_used(self, pspec=None, user_data=None):
if self.model_drop_down.get_selected_item() == None: return True if self.model_drop_down.get_selected_item() == None: return True
selected = self.model_drop_down.get_selected_item().get_string().split(":")[0] selected = self.model_drop_down.get_selected_item().get_string().split(":")[0]
if selected in ['llava']: if selected in ['llava', 'bakllava', 'moondream', 'llava-llama3']:
self.image_button.set_sensitive(True) self.image_button.set_sensitive(True)
return True return True
else: else:
@ -576,7 +576,8 @@ class AlpacaWindow(Adw.ApplicationWindow):
self.available_model_list_box.remove_all() self.available_model_list_box.remove_all()
for name, model_info in available_models.items(): for name, model_info in available_models.items():
model = Adw.ActionRow( model = Adw.ActionRow(
title = name title = name,
subtitle = "Image recognition" if model_info["image"] else None
) )
link_button = Gtk.Button( link_button = Gtk.Button(
icon_name = "web-browser-symbolic", icon_name = "web-browser-symbolic",