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

Fixed chat title generation and image recognition (+ added support for jpg)

Browse Source
This commit is contained in:
jeffser
2024-06-30 14:43:12 -06:00
parent c6110e8cde
commit b6a180080f
3 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/window.py
View File

@@ -156,8 +156,8 @@ class AlpacaWindow(Adw.ApplicationWindow):
@Gtk.Template.Callback()
def verify_if_image_can_be_used(self, pspec=None, user_data=None):
if self.model_drop_down.get_selected_item() == None: return True
selected = self.model_drop_down.get_selected_item().get_string().split(" (")[0]
if selected in ['llava', 'bakllava', 'moondream', 'llava-llama3']:
selected = self.model_drop_down.get_selected_item().get_string().split(" (")[0].lower()
if selected in [key for key, value in self.available_models.items() if value["image"]]:
for name, content in self.attachments.items():
if content['type'] == 'image':
content['button'].set_css_classes(["flat"])
@@ -501,7 +501,7 @@ Generate a title following these rules:
{message}
```"""
current_model = self.model_drop_down.get_selected_item().get_string()
current_model = current_model.replace(' (', ':')[:-1]
current_model = current_model.replace(' (', ':')[:-1].lower()
response = connection_handler.simple_post(f"{connection_handler.url}/api/generate", data=json.dumps({"model": current_model, "prompt": prompt, "stream": False}))
new_chat_name = json.loads(response['text'])["response"].replace('"', '').replace("'", "")
new_chat_name = self.generate_numbered_name(new_chat_name, self.chats["chats"].keys())