Update app.py

app.py

@@ -36,7 +36,9 @@ def generate_text(usertitle, content, temperature, max_length, N=3):
         with torch.no_grad():
             outputs = rm_model(**inputs)
             logits = outputs.logits
-            return logits[0][1].item()
+            print("Logits shape:", logits.shape)  
+            print("Logits contents:", logits) 
+            return logits[0]#[0].item()
 
     best_sequence = max(decoded_sequences, key=score)