a-ragab-h-m commited on
Commit
7bab133
·
verified ·
1 Parent(s): a370b55

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -33,7 +33,7 @@ device = params['device']
33
 
34
  # --- تحميل النموذج ---
35
  model = Model(
36
- input_size=4, # استخدام 4 خصائص فقط (lng1, lat1, lng2, lat2)
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
  )