Spaces:
Build error
Build error
| import numpy as np | |
| import gradio as gr | |
| import tensorflow as tf | |
| from transformers import TFAutoModelForSequenceClassification, AutoTokenizer | |
| checkpoint="manuu01/DeBERTa-SeagullStory" | |
| tokenizer = AutoTokenizer.from_pretrained(checkpoint) | |
| model=TFAutoModelForSequenceClassification.from_pretrained(checkpoint,from_pt=True,dtype=tf.float16) | |
| answers=['Yes',"Doesn't matter","No"] | |
| f=open("Questions.txt","a") | |
| title = "The Seagull story" | |
| description = """ | |
| Albert and Dave are at a pier. They go to a restaurant. Albert orders seagull. He takes a bite, then he kills himself.\n | |
| This is the final part of a story. It does not make sense at first, but in the end you can be sure it will actually make sense.\n | |
| You can ask boolean questions to which you will receive one of the following answers:\n | |
| "Yes" if it is true\n | |
| "No" if it is not true\n | |
| "Doesn't matter" if what you're asking is not important to understand the whole story.\n\n | |
| Example: Bob liked the restaurant\n | |
| --Answer: Doesn't matter\n | |
| Bob shot himself because he did not have money to pay\n | |
| --Answer: No | |
| """ | |
| def generate_tone(index,question): | |
| passages=[ | |
| """ | |
| Albert Dave and lucy are friends. they are completely normal people. Albert and lucy are happily married. Albert and lucy are | |
| loyal and really love each other, their love is true. Albert loves lucy. lucy loves Albert. Albert Dave and lucy are middle aged. | |
| Albert is blind. Albert is not religious. | |
| the story is set in the early 900's. the story is completely plausible and does not involve anything not realistic. | |
| """.replace("\n",""), | |
| """ | |
| A kind sailor saved Albert and Dave from an island and brought them to a pier with his boat. Now, they are sad and shocked. | |
| Albert does not know lucy is dead, Dave knows Lucy is dead. They go to a restaurant. Albert orders seagull meat. | |
| The waiter brings Albert the seagull meat. Everything is ordinary. The seagull meat is perfectly fine and edible. | |
| After Albert takes a bite, he realizes he had never eaten seagull meat before and because of that Albert kills himself. | |
| Dave knew Albert would suicide, so he is not surpirsed. Other people at the restaurant were surprised. | |
| Albert and Dave were exhausted. Albert and Dave had discussions. Albert wanted to go to the restaurant | |
| in order to eat seagull, Albert had this plan when he was on the island and nothing that happened changed Albert's mind. | |
| """.replace("\n",""), | |
| """ | |
| Albert Dave and lucy are having fun. they did not have any argument. | |
| what happened was not upon them. there is a storm and the boat is wrecked. Albert, Dave, lucy cast away on an island. | |
| """.replace("\n",""), | |
| """ | |
| lucy dies in the shipwreck. Albert and Dave manage to survive. they are alone on the island. there is no one else on the island. | |
| there are not dangerous animals. Albert and Dave do not have to fight to survive. there is no one else on the island. | |
| there is not much food.they resort to unconvential food sources. they did not have to fight. Dave finds lucy's corpse. | |
| they manage to survive.Dave is compelled to eat lucy. Dave does not want to eat lucy. Dave eats lucy. Albert also eats lucy. | |
| because Albert is blind, Albert does not know he is eating lucy. Dave knows he is eating lucy. Dave revolts while eating lucy. | |
| Albert does not revolt while eating lucy. Dave tells Albert he is eating seagull. Albert thinks he is eating seagull. | |
| Albert is suspicious. there is a sailor passing by. the sailor saves Albert and Dave and brings them to a pier. | |
| the sailor is a good man. the sailor does not have ill intentions. Albert and Dave are grateful to the sailor. | |
| Albert does not know lucy is dead. Dave knows lucy is dead. | |
| """.replace("\n","") | |
| ] | |
| inputs=tokenizer(passages[index],question, return_tensors="tf") | |
| response=(model(**inputs).logits) | |
| return model.config.id2label[np.argmax(response)] | |
| passages=["General","Pier","Boat","Island"] | |
| gr.Interface( | |
| generate_tone, | |
| [ | |
| gr.Dropdown(passages, type="index", value="Pier"), | |
| gr.Textbox(type="text", label="enter your question"), | |
| ], | |
| "text", | |
| title=title, | |
| description=description, | |
| ).launch() |