Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -14,8 +14,8 @@ import torch
|
|
14 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
15 |
|
16 |
# Загрузка модели и токенизатора
|
17 |
-
tokenizer = GPT2Tokenizer.from_pretrained("ERmak1581/
|
18 |
-
model = GPT2LMHeadModel.from_pretrained("ERmak1581/
|
19 |
|
20 |
# Функция для генерации текста на основе входной строки
|
21 |
def gen(request, temperature, maxnewtokens):
|
@@ -49,5 +49,5 @@ inputs = [
|
|
49 |
]
|
50 |
output = gr.Textbox(label="Output Text")
|
51 |
|
52 |
-
interface = gr.Interface(gen, inputs, output, title="GPT-2 Text Generation", theme="compact", description="Демонстрация <a href=https://huggingface.co/ERmak1581/
|
53 |
interface.launch(share=True)
|
|
|
14 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
15 |
|
16 |
# Загрузка модели и токенизатора
|
17 |
+
tokenizer = GPT2Tokenizer.from_pretrained("ERmak1581/rugpt3large_for_qna_400k")
|
18 |
+
model = GPT2LMHeadModel.from_pretrained("ERmak1581/rugpt3large_for_qna_400k")
|
19 |
|
20 |
# Функция для генерации текста на основе входной строки
|
21 |
def gen(request, temperature, maxnewtokens):
|
|
|
49 |
]
|
50 |
output = gr.Textbox(label="Output Text")
|
51 |
|
52 |
+
interface = gr.Interface(gen, inputs, output, title="GPT-2 Text Generation", theme="compact", description="Демонстрация <a href=https://huggingface.co/ERmak1581/rugpt3large_for_qna_400k>модели</a> для задачи Question-Answer")
|
53 |
interface.launch(share=True)
|