Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -33,7 +33,7 @@ device = params['device']
|
|
33 |
|
34 |
# --- تحميل النموذج ---
|
35 |
model = Model(
|
36 |
-
input_size=
|
37 |
embedding_size=params["embedding_size"],
|
38 |
decoder_input_size=params["decoder_input_size"]
|
39 |
)
|
|
|
33 |
|
34 |
# --- تحميل النموذج ---
|
35 |
model = Model(
|
36 |
+
input_size = 5,
|
37 |
embedding_size=params["embedding_size"],
|
38 |
decoder_input_size=params["decoder_input_size"]
|
39 |
)
|