luna-playground

Sleeping

terryyz commited on Mar 12

Commit

9b1a775

•

1 Parent(s): f8c3014

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -182,7 +182,7 @@ def generate(
     else:
         output = ""
     # model.to(device)
-    input_ids = tokenizer(code, return_tensors="pt").to(device)
     # generated_ids = model.generate(**input_ids, **generate_kwargs)
     generated_ids = model.generate(**input_ids)

     else:
         output = ""
     # model.to(device)
+    input_ids = tokenizer(prompt, return_tensors="pt").to(device)
     # generated_ids = model.generate(**input_ids, **generate_kwargs)
     generated_ids = model.generate(**input_ids)