AlexWortega commited on
Commit
cdc5960
1 Parent(s): 0cac0a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ tokenizer = GPT2TokenizerFast.from_pretrained("AlexWortega/instruct_rugptlarge")
9
  special_tokens_dict = {'additional_special_tokens': ['<code>', '</code>', '<instructionS>', '<instructionE>', '<next>']}
10
 
11
  tokenizer.add_special_tokens(special_tokens_dict)
12
- device = 'cuda' # мэх дорога
13
  model = GPT2LMHeadModel.from_pretrained("AlexWortega/instruct_rugptlarge")
14
  #
15
 
@@ -96,4 +96,4 @@ g = gr.Interface(
96
 
97
 
98
  g.queue(concurrency_count=5)
99
- g.launch(share=True)
 
9
  special_tokens_dict = {'additional_special_tokens': ['<code>', '</code>', '<instructionS>', '<instructionE>', '<next>']}
10
 
11
  tokenizer.add_special_tokens(special_tokens_dict)
12
+ device = 'cpu' # мэх дорога
13
  model = GPT2LMHeadModel.from_pretrained("AlexWortega/instruct_rugptlarge")
14
  #
15
 
 
96
 
97
 
98
  g.queue(concurrency_count=5)
99
+ g.launch()