YasinOnm08/ai-virtual-assistant
1
0
Fork 0
forked from React-Group/ai-virtual-assistant
Code Pull requests Activity

Fixed html.

Browse source
This commit is contained in:
Patrick_Pluto 2024-09-16 13:38:30 +02:00
parent 4cbb44b9fc
commit e3cdde8a74
3 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
py/api.py
View file

@ -2,6 +2,7 @@ import requests
import json
from transformers import AutoTokenizer, LlamaForCausalLM
class API:
@staticmethod
def process_text_transformers(prompt, model):
@ -46,4 +47,4 @@ class API:
answer = self.process_text_transformers(message, "meta-llama/Meta-Llama-3.1-8B")
else:
return "Invalid choice"
return answer
return answer