jrosh/LectureSummarizer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update llm_page.py

Browse source
This commit is contained in:
Martin Jaros 2024-04-25 20:43:23 +02:00
commit 14877238e0
1 changed files with 1 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lsb_package/llm_page.py
View file

@ -29,10 +29,8 @@ class LlmPage(tk.Frame):
def run_llama_operation(self, llmTopics): def run_llama_operation(self, llmTopics):
try: try:
print("##### Started Llama...")
print(llmTopics)
# Example: Llama class must be imported correctly here # Example: Llama class must be imported correctly here
llm = Llama(model_path="/home/jrosh/.local/share/nomic.ai/GPT4All/gpt4all-falcon-newbpe-q4_0.gguf", n_ctx=2048, ) llm = Llama(model_path="##############", n_ctx=2048, )
# output = llm.create_chat_completion( # output = llm.create_chat_completion(
# messages=[ # messages=[
# {"role": "system", "content": "You are a teacher explaining in great detail given topics divided by new line."}, # {"role": "system", "content": "You are a teacher explaining in great detail given topics divided by new line."},
@ -79,10 +77,7 @@ class LlmPage(tk.Frame):
Returns: Returns:
Response object containing the generated text. Response object containing the generated text.
""" """
print(output)
#print(output['choices'][0]['message']['content'])
self.text_widget.after(0, self.update_text_widget, output['choices'][0]) self.text_widget.after(0, self.update_text_widget, output['choices'][0])
print("##### Llama Finished")
except Exception as e: except Exception as e:
print(f"Error during Llama operation: {e}") print(f"Error during Llama operation: {e}")
self.text_widget.after(0, self.update_text_widget, "An error occurred, please try again.") self.text_widget.after(0, self.update_text_widget, "An error occurred, please try again.")