Upload 2 files

app.py

@@ -0,0 +1,10 @@
+import gradio as gr
+from transformers import pipeline
+
+pipe = pipeline("text2text-generation", model="yxccai/text-style")
+
+def chat(input_text):
+    result = pipe(input_text, max_new_tokens=512)[0]["generated_text"]
+    return result
+
+gr.Interface(fn=chat, inputs="text", outputs="text").launch()

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+gradio