Upload app.py

app.py

@@ -92,8 +92,10 @@ class DrugTargetInteractionApp:
                 max_length=512,
                 return_tensors="pt"
             ).to(self.device)
-            
-            print(drug_inputs.shape, target_inputs.shape)
+
+            logger.info(f"Drug inputs: {drug_inputs}")
+            logger.info(f"Target inputs: {target_inputs}")
+
             # Make prediction
             with torch.no_grad():
                 prediction = self.model(target_inputs, drug_inputs)