Spaces:
Paused
Paused
DHRUV SHEKHAWAT
commited on
Commit
·
ff1e41f
1
Parent(s):
130d634
Update app.py
Browse files
app.py
CHANGED
@@ -127,18 +127,7 @@ elif option=="26M":
|
|
127 |
len = 1
|
128 |
else:
|
129 |
len = 13
|
130 |
-
|
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 |
|