ThanaritKanjanametawat
Move everything to CPU
1bb143a
raw
history blame
877 Bytes
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))