DHRUV SHEKHAWAT commited on
Commit
ff1e41f
·
1 Parent(s): 130d634

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -12
app.py CHANGED
@@ -127,18 +127,7 @@ elif option=="26M":
127
  len = 1
128
  else:
129
  len = 13
130
- vocab_size = 100000
131
- max_len = 1
132
- d_model = 128 # 64 , 1024
133
- n_head = 8 # 8 , 16
134
- ff_dim = 256 # 256 , 2048
135
- dropout_rate = 0.1 # 0.5 , 0.2
136
- weights = "predict5"
137
- datafile = "data2.txt"
138
- dict = "dict_predict3.bin.npz"
139
- out2 = completion_model(vocab_size, max_len, d_model, n_head, ff_dim, dropout_rate,weights,datafile,dict,len,text2)
140
- st.write("Predicted Text: ")
141
- st.write(out2)
142
  else:
143
  out2 = "Error: Wrong Model Selected"
144
 
 
127
  len = 1
128
  else:
129
  len = 13
130
+
 
 
 
 
 
 
 
 
 
 
 
131
  else:
132
  out2 = "Error: Wrong Model Selected"
133