Update inference.py

inference.py

@@ -207,7 +207,7 @@ class RadFigClassifier:
             'image_path': image_paths,
             'filename': [os.path.basename(path) for path in image_paths],
             'prediction': predictions,
-            'suitable_for_vqa': predictions > 0.5 if return_probabilities else predictions.astype(bool)
+            'suitable_for_vqa': predictions > 0.9 if return_probabilities else predictions.astype(bool)
         })
         
         # Sort by filename for consistency
@@ -250,7 +250,7 @@ def main():
         else:
             print(f"Image: {args.input}")
             print(f"Probability suitable for VQA: {prediction:.4f}")
-            print(f"Classification: {'suitable' if prediction > 0.5 else 'not suitable'}")
+            print(f"Classification: {'suitable' if prediction > 0.9 else 'not suitable'}")
             
     elif os.path.isdir(args.input):
         # Directory prediction