ElijahDi commited on
Commit
c816ee9
·
verified ·
1 Parent(s): 1f92bd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -10,7 +10,6 @@ import joblib
10
  from dataclasses import dataclass
11
  from transformers import AutoTokenizer, AutoModel, AutoModelForSequenceClassification, GPT2LMHeadModel, GPT2Tokenizer
12
  import json
13
- import os
14
 
15
  from preprocessing import predict_review, data_preprocessing_hard
16
  from model_lstm import LSTMClassifier
@@ -188,9 +187,8 @@ elif selected_model == "Генерация текста GPT-моделью по
188
 
189
  # GPT2
190
  model_path = "model.safetensors"
191
- huggingface_token = os.getenv("HF_TOKEN")
192
- model = GPT2LMHeadModel.from_pretrained(model_path, token=huggingface_token)
193
- tokenizer = GPT2Tokenizer.from_pretrained(model_path, token=huggingface_token)
194
 
195
  if st.button('Сделать гороскоп'):
196
  start_time = time.time()
 
10
  from dataclasses import dataclass
11
  from transformers import AutoTokenizer, AutoModel, AutoModelForSequenceClassification, GPT2LMHeadModel, GPT2Tokenizer
12
  import json
 
13
 
14
  from preprocessing import predict_review, data_preprocessing_hard
15
  from model_lstm import LSTMClassifier
 
187
 
188
  # GPT2
189
  model_path = "model.safetensors"
190
+ model = GPT2LMHeadModel.from_pretrained(model_path)
191
+ tokenizer = GPT2Tokenizer.from_pretrained(model_path)
 
192
 
193
  if st.button('Сделать гороскоп'):
194
  start_time = time.time()