ERmak1581 commited on
Commit
feb79c2
·
verified ·
1 Parent(s): 27e247b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,8 +14,8 @@ import torch
14
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
15
 
16
  # Загрузка модели и токенизатора
17
- tokenizer = GPT2Tokenizer.from_pretrained("ERmak1581/rugpt3large_for_qna_250k1")
18
- model = GPT2LMHeadModel.from_pretrained("ERmak1581/rugpt3large_for_qna_250k1")
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/rugpt3large_for_qna_250k1>модели</a> для задачи Question-Answer")
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)