Update app.py
Browse files
app.py
CHANGED
|
@@ -166,7 +166,7 @@ def Generate_POP_Section(input_comp_section,
|
|
| 166 |
|
| 167 |
seq += input_seq
|
| 168 |
|
| 169 |
-
x = torch.LongTensor(seq).
|
| 170 |
|
| 171 |
with ctx:
|
| 172 |
out = model.generate(x,
|
|
|
|
| 166 |
|
| 167 |
seq += input_seq
|
| 168 |
|
| 169 |
+
x = torch.LongTensor(seq).to(DEVICE)
|
| 170 |
|
| 171 |
with ctx:
|
| 172 |
out = model.generate(x,
|