fsd

app.py

@@ -1,7 +1,16 @@
 import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-def greet(name):
-    return "Hello " + name + "!!"
+tokenizer = AutoTokenizer.from_pretrained("Gorilla115/t5-shakespearify-lite")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+model = AutoModelForSeq2SeqLM.from_pretrained("Gorilla115/t5-shakespearify-lite")
+
+def Generate(text):
+    enc= tokenizer.encode("translate: "+text,return_tensors="pt")
+    out = model.generate(enc)
+    return tokenizer.decode(out[0]).replace("<pad>","").replace("</s>","")
+
+    # return "Hello " + name + "!!"
+
+iface = gr.Interface(fn=Generate, inputs="text", outputs="text",examples=["Why have you come to Mr. Smith with this crap?","Who are you man?","You are in need of great care to fix this.","That’s true, and he asked me to beg both of you, your Majesties, to come and watch."])
 iface.launch()

requirements.txt

@@ -1 +1,50 @@
-gradio
+aiohttp==3.8.1
+aiosignal==1.2.0
+analytics-python==1.4.0
+anyio==3.6.1
+async-timeout==4.0.2
+backoff==1.10.0
+bcrypt==3.2.2
+click==8.1.3
+cycler==0.11.0
+fastapi==0.79.1
+ffmpy==0.3.0
+fonttools==4.36.0
+frozenlist==1.3.1
+fsspec==2022.7.1
+gradio==3.1.6
+h11==0.12.0
+httpcore==0.15.0
+httpx==0.23.0
+huggingface-hub==0.8.1
+kiwisolver==1.4.4
+linkify-it-py==1.0.3
+markdown-it-py==2.1.0
+matplotlib==3.5.3
+mdit-py-plugins==0.3.0
+mdurl==0.1.2
+monotonic==1.6
+multidict==6.0.2
+numpy==1.23.2
+orjson==3.7.12
+pandas==1.4.3
+paramiko==2.11.0
+Pillow==9.2.0
+pycryptodome==3.15.0
+pydantic==1.9.2
+pydub==0.25.1
+PyNaCl==1.5.0
+python-multipart==0.0.5
+pytz==2022.2.1
+regex==2022.8.17
+rfc3986==1.5.0
+sniffio==1.2.0
+starlette==0.19.1
+tokenizers==0.12.1
+torch==1.12.1
+tqdm==4.64.0
+transformers==4.21.1
+uc-micro-py==1.0.1
+uvicorn==0.18.2
+websockets==10.3
+yarl==1.8.1