VeraMind/main.py

16 lines
311 B
Python
Raw Permalink Normal View History

2024-09-18 20:12:00 +00:00
from src.Inference import VeraMindInference
# load model
2024-12-29 20:48:18 +00:00
model = VeraMindInference("VeraMind-Edge")
2024-09-18 20:12:00 +00:00
text = "This is a example News Article"
# predict if News are reel or Fake
result = model.predict(text)
# Example Output
2024-12-29 20:48:18 +00:00
# {'result': 'FAKE', 'confidence': 0.9981970191001892}
2024-09-18 20:12:00 +00:00
print(result)