Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
f4d6d33
1
Parent(s):
5fec312
Update app.py
Browse files
app.py
CHANGED
|
@@ -160,6 +160,9 @@ def Generate_POP_Section(input_comp_section,
|
|
| 160 |
if 0 in input_seq:
|
| 161 |
input_seq = input_seq_toks[:input_seq.index(0)]
|
| 162 |
|
|
|
|
|
|
|
|
|
|
| 163 |
seq += input_seq
|
| 164 |
|
| 165 |
model.to(DEVICE)
|
|
|
|
| 160 |
if 0 in input_seq:
|
| 161 |
input_seq = input_seq_toks[:input_seq.index(0)]
|
| 162 |
|
| 163 |
+
else:
|
| 164 |
+
input_seq = input_seq_toks
|
| 165 |
+
|
| 166 |
seq += input_seq
|
| 167 |
|
| 168 |
model.to(DEVICE)
|