project_charles / debug.py
sohojoe's picture
prep to test on machine with gpu
35d97c8
raw
history blame
622 Bytes
from clip_transform import CLIPTransform
from chat_service import ChatService
from dotenv import load_dotenv
load_dotenv()
print ("Initializing CLIP templates")
clip_transform = CLIPTransform()
print ("CLIP success")
print ("Initializing Chat")
chat_service = ChatService()
prompts = [
"hello, how are you today?",
"tell me about your showdow self?",
"hmm, interesting, tell me more about that.",
"wait, that is so interesting, what else?",
]
for prompt in prompts:
print (f'prompt: "{prompt}"')
response = chat_service.chat(prompt)
print (f'response: "{response}"')
print ("Chat success")