Spaces:

ajimenez78
/

python2cplusplus

Paused

Arturo Jiménez de los Galanes Reguillos commited on Nov 4, 2024

Commit

87c9e01

1 Parent(s): b9dbd4f

Remove chat template usage

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ streamer = TextIteratorStreamer(tokenizer)
 cplusplus = None
 def translate(python):
-    inputs = tokenizer.apply_chat_template(messages_for(python), return_tensors="pt").to(model.device)
     generation_kwargs = dict(
                             inputs,
                             streamer=streamer,

 cplusplus = None
 def translate(python):
+    inputs = tokenizer(messages_for(python), return_tensors="pt").to(model.device)
     generation_kwargs = dict(
                             inputs,
                             streamer=streamer,