Update app.py

app.py

@@ -161,7 +161,6 @@ def generate(
         top_p=top_p,
         repetition_penalty=repetition_penalty,
         do_sample=True,
-        seed=42,
     )
 
     if method == "Base":
@@ -182,10 +181,9 @@ def generate(
         output = ""
     model.to(device)
     input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
-    # generated_ids = model.generate(**input_ids
     generated_ids = model.generate(**input_ids, **generate_kwargs)
-    
-    return tokenizer.decode(generated_ids[0][input_ids["input_ids"].shape[1]:], skip_special_tokens=True).strip()
+    return tokenizer.decode(generated_ids[0], skip_special_tokens=True).strip()
+    # return tokenizer.decode(generated_ids[0][input_ids["input_ids"].shape[1]:], skip_special_tokens=True).strip()
 
 
 examples = [