Spaces:
Paused
Paused
Kobkrit Viriyayudhakorn
commited on
Commit
•
cfce0d9
1
Parent(s):
c09e81b
Add OpenThaiGPT Zero
Browse files- app.py +13 -15
- requirements.txt +1 -4
app.py
CHANGED
@@ -1,21 +1,19 @@
|
|
1 |
import gradio as gr
|
2 |
-
|
3 |
-
|
4 |
-
import torch
|
5 |
-
from torch.utils.data import Dataset, random_split
|
6 |
-
from transformers import GPT2Tokenizer, TrainingArguments, Trainer, GPT2LMHeadModel
|
7 |
|
8 |
-
|
|
|
9 |
|
10 |
-
|
11 |
-
|
12 |
-
model.resize_token_embeddings(len(tokenizer))
|
13 |
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
|
|
|
|
19 |
|
20 |
-
demo = gr.Interface(fn=gen, inputs=gr.Textbox(lines=3, label="Input Text", value="Q: อยากลดความอ้วน ทำอย่างไร\n\nA:"), outputs="text")
|
21 |
demo.launch()
|
|
|
1 |
import gradio as gr
|
2 |
+
import openthaigpt
|
3 |
+
demo = gr.Blocks()
|
|
|
|
|
|
|
4 |
|
5 |
+
def gen(input):
|
6 |
+
return openthaigpt.generate(input)
|
7 |
|
8 |
+
def zero(input):
|
9 |
+
return openthaigpt.zero(input)
|
|
|
10 |
|
11 |
+
with demo:
|
12 |
+
gr.Markdown("OpenThaiGPT version 0.0.10")
|
13 |
+
with gr.Tabs():
|
14 |
+
with gr.TabItem("Generate"):
|
15 |
+
gr.Interface(fn=gen, inputs=gr.Textbox(lines=3, label="Input Prompt", value="Q: อยากลดความอ้วน ทำอย่างไร\n\nA:"), outputs="text")
|
16 |
+
with gr.TabItem("Zero (GPT Check)"):
|
17 |
+
gr.Interface(fn=zero, inputs=gr.Textbox(lines=3, label="Input Text", value="การลดน้ำหนักเป็นเรื่องที่ต้องพิจารณาอย่างละเอียดและรอบคอบเพื่อให้ได้ผลลัพธ์ที่ดีและมีประสิทธิภาพมากที่สุด"), outputs="text")
|
18 |
|
|
|
19 |
demo.launch()
|
requirements.txt
CHANGED
@@ -1,4 +1 @@
|
|
1 |
-
|
2 |
-
datasets
|
3 |
-
#--extra-index-url https://download.pytorch.org/whl/cpu
|
4 |
-
torch==1.13.1
|
|
|
1 |
+
openthaigpt
|
|
|
|
|
|