Fix attempt 2

app.py

@@ -22,11 +22,11 @@ rm_model = AutoModelForSequenceClassification.from_pretrained('OpenAssistant/rew
 
 @spaces.GPU
 def generate_text(usertitle, content, temperature, max_length, N=3):
-    # msg ={
-    # 'title': usertitle,
-    # 'content': content
-    # }
-    input_ids = tokenizer.apply_chat_template(input_text, return_tensors='pt').to('cuda')
+     msg = [{
+     'title': usertitle,
+     'content': content
+    }]
+    input_ids = tokenizer.apply_chat_template(msg, return_tensors='pt').to('cuda')
     attention_mask = torch.ones(input_ids.shape, dtype=torch.long, device='cuda')
     generated_sequences = model.generate(input_ids, attention_mask=attention_mask, temperature=temperature, max_length=max_length, pad_token_id=tokenizer.eos_token_id, num_return_sequences=N, do_sample=True)
     decoded_sequences = [tokenizer.decode(g, skip_special_tokens=True) for g in generated_sequences]