franz96521 commited on
Commit
ed36ce1
·
1 Parent(s): d690839
Files changed (2) hide show
  1. app.py +1 -5
  2. requirements.txt +2 -0
app.py CHANGED
@@ -1,11 +1,10 @@
1
  import os
2
  os.system('pip install gpt-2-simple')
 
3
  os.system('pip install tensorflow-estimator==1.15.1')
4
 
5
  import gpt_2_simple as gpt2
6
  import tensorflow as tf
7
- import pandas as pd
8
- import re
9
  import gradio as gr
10
 
11
  model_name = "124M"
@@ -15,9 +14,6 @@ checkpoint_dir =path+'weights/'
15
  data_path = path+'Tokenized_data/'
16
 
17
 
18
- file_name = 'resumen'
19
- file_path = data_path+file_name
20
-
21
  prefix= '<|startoftext|>'
22
  sufix ='<|endoftext|>'
23
 
 
1
  import os
2
  os.system('pip install gpt-2-simple')
3
+ os.system('pip install tensorflow-gpu==1.15')
4
  os.system('pip install tensorflow-estimator==1.15.1')
5
 
6
  import gpt_2_simple as gpt2
7
  import tensorflow as tf
 
 
8
  import gradio as gr
9
 
10
  model_name = "124M"
 
14
  data_path = path+'Tokenized_data/'
15
 
16
 
 
 
 
17
  prefix= '<|startoftext|>'
18
  sufix ='<|endoftext|>'
19
 
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ tensorflow-gpu==1.15
2
+ gpt_2_simple