from ModelDriver import *
import numpy as np
import warnings
warnings.filterwarnings("ignore")

Input_Text = "I want to do this data"

# print("RobertaSentinelOpenGPTInference")
# Probs = RobertaSentinelOpenGPTInference(Input_Text)
# Pred = "Human Written" if not np.argmax(Probs) else "Machine Generated"
#
# print(f"Prediction: {Pred} ")
# print(f"Confidence:", max(Probs))

# print("RobertaSentinelCSAbstractInference")
# Probs = RobertaSentinelCSAbstractInference(Input_Text)
# Pred = "Human Written" if not np.argmax(Probs) else "Machine Generated"
#
# print(f"Prediction: {Pred} ")
# print(f"Confidence:", max(Probs))

print("RobertaClassifierCSAbstractInference")
Probs = RobertaClassifierCSAbstractInference(Input_Text)
Pred = "Human Written" if not np.argmax(Probs) else "Machine Generated"

print(Probs)
print(f"Prediction: {Pred} ")
print(f"Confidence:", max(Probs))