File size: 877 Bytes
bd0c703
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1bb143a
bd0c703
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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))