Update app.py

app.py

@@ -22,10 +22,10 @@ rm_model = AutoModelForSequenceClassification.from_pretrained('OpenAssistant/rew
 
 @spaces.GPU
 def generate_text(usertitle, content, temperature, max_length, N=3):
-    msg =[
-    'title': {usertitle},
-    'content': {content}
-    ]
+    msg ={
+    'title': usertitle,
+    'content': content
+    }
     # input_text = f"title: {usertitle}\ncontent: {content}"
     inputs = tokenizer.apply_chat_template(msg, return_tensors='pt').cuda()
     attention_mask = torch.ones(inputs['input_ids'].shape, dtype=torch.long, device='cuda')