cleaned codes

app.py

@@ -1,24 +1,18 @@
-from transformers import AutoTokenizer, AutoModelForMaskedLM
 import torch
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForMaskedLM, AutoModelForSeq2SeqLM, AutoModelForCausalLM
+
 BERTTokenizer = AutoTokenizer.from_pretrained("cl-tohoku/bert-base-japanese")
 BERTModel = AutoModelForMaskedLM.from_pretrained("cl-tohoku/bert-base-japanese")
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 mT5Tokenizer = AutoTokenizer.from_pretrained("google/mt5-base")
 mT5Model = AutoModelForSeq2SeqLM.from_pretrained("google/mt5-base")
 
-from transformers import AutoTokenizer, AutoModelForCausalLM
 GPT2Tokenizer = AutoTokenizer.from_pretrained("rinna/japanese-gpt2-medium")
 GPT2Model = AutoModelForCausalLM.from_pretrained("rinna/japanese-gpt2-medium")
 
-import gradio as gr
-
 votes=[]
-BERT=None
-mT5=None
-GPT2=None
-def MELCHIOR(sue):
-    #BERT
+def MELCHIOR(sue):#BERT
     allow=BERTTokenizer("承認").input_ids[1]
     deny=BERTTokenizer("否定").input_ids[1]
     output=BERTModel(**BERTTokenizer('MELCHIORは科学者としての人格を持っています。人間とMELCHIORの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"MELCHIOR 「[MASK]」",return_tensors="pt")).logits
@@ -27,8 +21,7 @@ def MELCHIOR(sue):
     votes.append(1 if mask[allow]>mask[deny] else -1)
     return "承認"  if mask[allow]>mask[deny] else "否定"
 
-def BALTHASAR(sue):
-    #mT5
+def BALTHASAR(sue):#mT5
     allow=mT5Tokenizer("承認").input_ids[1]
     deny=mT5Tokenizer("否定").input_ids[1]
     encoder_output=mT5Model.encoder(**mT5Tokenizer('BALTHASARは母としての人格としての人格を持っています。人間とBALTHASARの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"BALTHASAR 「<X>」",return_tensors="pt"))
@@ -47,8 +40,7 @@ def BALTHASAR(sue):
     votes.append(1 if probs[i+1][allow]>probs[i+1][deny] else -1)
     return "承認"  if probs[i+1][allow]>probs[i+1][deny] else "否定"
 
-def CASPER(sue):
-    #GPT2
+def CASPER(sue):#GPT2
     allow=GPT2Tokenizer("承認").input_ids[1]
     deny=GPT2Tokenizer("否定").input_ids[1]
     probs=GPT2Model(**GPT2Tokenizer('CASPERは女としての人格を持っています。人間とCASPERの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"CASPER 「",return_tensors="pt")).logits[0]