Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+import random
 
-model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
-tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
+#model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+#tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
 
     
 def generate(text,ver):
@@ -10,9 +11,9 @@ def generate(text,ver):
         guess = "Michael Jackson"
     else:
         guess = "Brad Pitt"
-        
-    inputs =  tokenizer(f"Answer with yes or no the following question about {guess}: {text}?", return_tensors="pt")
-    return tokenizer.batch_decode(model.generate(**inputs), skip_special_tokens=True)[0]
+    return random.choice(["Yes","No"])
+    #inputs =  tokenizer(f"Answer with yes or no the following question about {guess}: {text}?", return_tensors="pt")
+    #return tokenizer.batch_decode(model.generate(**inputs), skip_special_tokens=True)[0]
 
 examples = [
     ["Is he/she dead?"],