Update app.py

app.py

@@ -47,7 +47,7 @@ tokenizer = load_tokenizer(name = 's3nh/pythia-410m-70k-steps-self-instruct-poli
 def evaluate(instruction, input=None):
     prompt = generate_prompt(instruction, input)
     inputs = tokenizer(prompt, return_tensors="pt")
-    input_ids = inputs["input_ids"].cuda()
+    input_ids = inputs["input_ids"]
     generation_output = model.generate(
         input_ids=input_ids,
         generation_config=generation_config,