Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -179,7 +179,7 @@ if st.button('확인'):
|
|
| 179 |
logits = outputs.logits
|
| 180 |
|
| 181 |
# Move logits and labels to CPU
|
| 182 |
-
logits = logits.detach().cpu().numpy()
|
| 183 |
|
| 184 |
|
| 185 |
# # 단독 예측 시
|
|
|
|
| 179 |
logits = outputs.logits
|
| 180 |
|
| 181 |
# Move logits and labels to CPU
|
| 182 |
+
# logits = logits.detach().cpu().numpy()
|
| 183 |
|
| 184 |
|
| 185 |
# # 단독 예측 시
|