dd
Browse files
app.py
CHANGED
|
@@ -27,7 +27,8 @@ def get_model():
|
|
| 27 |
|
| 28 |
model = get_model()
|
| 29 |
|
| 30 |
-
model.load_state_dict(model_weights)
|
|
|
|
| 31 |
|
| 32 |
import requests
|
| 33 |
from PIL import Image
|
|
|
|
| 27 |
|
| 28 |
model = get_model()
|
| 29 |
|
| 30 |
+
model.load_state_dict(model_weights)
|
| 31 |
+
model.eval()
|
| 32 |
|
| 33 |
import requests
|
| 34 |
from PIL import Image
|