VeraMind/main.py

16 lines
319 B
Python
Raw Permalink Normal View History

2024-09-18 20:12:00 +00:00
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)