add logging

app.py

@@ -64,23 +64,35 @@ def run_evaluation(model_name):
         pred_path = Path(output_dir) / pred_filename
         results.append(f"Prediction will be saved to: {pred_path}")
 
+        # Debug: Print predict function signature
+        yield f"Predict function signature: {inspect.signature(predict)}"
+
         # Run prediction
-        results.append("Starting prediction...")
-        predict(
-            dataset_path=dataset_path,
-            table_meta_path=table_meta_path,
-            output_dir=output_dir,
-            prompt_format=prompt_format,
-            stop_tokens=stop_tokens,
-            max_tokens=max_tokens,
-            temperature=temperature,
-            num_beams=num_beams,
-            manifest_client=manifest_client,
-            manifest_engine=manifest_engine,
-            manifest_connection=manifest_connection,
-            overwrite_manifest=overwrite_manifest,
-            parallel=parallel
-        )
+        yield "Starting prediction..."
+        try:
+            predict(
+                dataset_path=dataset_path,
+                table_meta_path=table_meta_path,
+                output_dir=output_dir,
+                prompt_format=prompt_format,
+                stop_tokens=stop_tokens,
+                max_tokens=max_tokens,
+                temperature=temperature,
+                num_beams=num_beams,
+                manifest_client=manifest_client,
+                manifest_engine=manifest_engine,
+                manifest_connection=manifest_connection,
+                overwrite_manifest=overwrite_manifest,
+                parallel=parallel
+            )
+        except TypeError as e:
+            yield f"TypeError in predict function: {str(e)}"
+            yield "Attempting to call predict with only expected arguments..."
+            # Try calling predict with only the arguments it expects
+            predict_args = inspect.getfullargspec(predict).args
+            filtered_args = {k: v for k, v in locals().items() if k in predict_args}
+            predict(**filtered_args)
+            
         results.append("Prediction completed.")
 
         # Run evaluation