Update app.py

app.py

@@ -408,7 +408,9 @@ def analyze_contract_clauses(text):
         ]
     chunks = [text[i:i+max_length] for i in range(0, len(text), step) if i+step < len(text)]
     for chunk in chunks:
-        inputs = cuad_tokenizer(chunk, return_tensors="pt", truncation=True, max_length=512).to(device)
+        # Move each tensor individually to the device
+        tokenized_inputs = cuad_tokenizer(chunk, return_tensors="pt", truncation=True, max_length=512)
+        inputs = {k: v.to(device) for k, v in tokenized_inputs.items()}
         with torch.no_grad():
             outputs = cuad_model(**inputs)
         predictions = torch.sigmoid(outputs.start_logits).cpu().numpy()[0]