VeraMind/main.py

16 lines
319 B
Python

from src.Inference import VeraMindInference
# load model
model = VeraMindInference("path/to/VeraMind-Mini")
text = "This is a example News Article"
# predict if News are reel or Fake
result = model.predict(text)
# Example Output
# {'result': 'FAKE', 'confidence': 0.9990140199661255}
print(result)