Spaces:

ddosxd
/

test-chat-zerogpu

Sleeping

ddosxd commited on Feb 5, 2024

Commit

462736e

verified ·

1 Parent(s): 65440f9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ zero = torch.Tensor([0]).cuda()
 @spaces.GPU
 def chat(prompt):
-    conversation = [ {'role': 'user', 'content': 'Hello?'} ]
     prompt = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)

 @spaces.GPU
 def chat(prompt):
+    conversation = [ {'role': 'user', 'content': prompt} ]
     prompt = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)