Duplicate from impira/flan-playground

Co-authored-by: Ankur Goyal <ankrgyl@users.noreply.huggingface.co>

.flake8

@@ -0,0 +1,3 @@
+[flake8]
+max-line-length = 119
+ignore = E402, E203, E501, W503

.gitignore

@@ -0,0 +1,5 @@
+venv
+*.swo
+*.swp
+*.pyc
+flagged_data_points

.isort.cfg

@@ -0,0 +1,6 @@
+[settings]
+line_length=119
+multi_line_output=3
+use_parentheses=true
+lines_after_imports=2
+include_trailing_comma=True

.pre-commit-config.yaml

@@ -0,0 +1,29 @@
+repos:
+  - repo: "https://github.com/pre-commit/pre-commit-hooks"
+    rev: v4.3.0
+    hooks:
+      - id: check-yaml
+      - id: end-of-file-fixer
+      - id: trailing-whitespace
+  - repo: "https://github.com/psf/black"
+    rev: 22.6.0
+    hooks:
+      - id: black
+        files: ./
+  - repo: "https://github.com/PyCQA/isort"
+    rev: 5.10.1
+    hooks:
+      - id: isort
+        args:
+          - --settings-path
+          - .isort.cfg
+        files: ./
+  - repo: https://github.com/codespell-project/codespell
+    rev: v2.2.1
+    hooks:
+      - id: codespell
+
+  - repo: https://github.com/pre-commit/mirrors-prettier
+    rev: v2.7.1
+    hooks:
+      - id: prettier

README.md

@@ -0,0 +1,11 @@
+---
+title: Flan Playground
+emoji: 🍮
+colorFrom: green
+colorTo: green
+sdk: gradio
+sdk_version: 3.2
+app_file: app.py
+pinned: true
+duplicated_from: impira/flan-playground
+---

app.py

@@ -0,0 +1,33 @@
+from transformers import pipeline
+import gradio as gr
+
+PIPELINES = {}
+
+
+def build_pipeline(size):
+    global PIPELINES
+    if size in PIPELINES:
+        return PIPELINES[size]
+
+    PIPELINES[size] = pipeline(
+        "text2text-generation", model=f"google/flan-t5-{size}", max_length=256
+    )
+    return PIPELINES[size]
+
+
+def greet(input_text, size):
+    pipe = build_pipeline(size)
+    return pipe(input_text)[0]["generated_text"]
+
+
+demo = gr.Interface(
+    fn=greet,
+    inputs=[
+        gr.Textbox(lines=2, placeholder="Enter your task text..."),
+        gr.Radio(choices=["small", "base", "large", "xl"], value="base"),
+    ],
+    outputs=[gr.Textbox(lines=2)],
+)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,2 @@
+torch
+transformers==4.22.2