Spaces:
Running
Running
altawil
commited on
Update model_definition.py
Browse files- model_definition.py +1 -1
model_definition.py
CHANGED
@@ -1243,7 +1243,7 @@ def get_master_config(model_path="model/best_model.pth"):
|
|
1243 |
return master_config
|
1244 |
|
1245 |
|
1246 |
-
def load_and_prepare_model(device: torch.device) ->
|
1247 |
"""
|
1248 |
[النسخة النهائية الصحيحة - تستقبل مدخلاً واحدًا فقط]
|
1249 |
تستخدم دالة الإعدادات الرئيسية لإنشاء وتحميل النموذج.
|
|
|
1243 |
return master_config
|
1244 |
|
1245 |
|
1246 |
+
def load_and_prepare_model(device: torch.device) -> InterfuserModel:
|
1247 |
"""
|
1248 |
[النسخة النهائية الصحيحة - تستقبل مدخلاً واحدًا فقط]
|
1249 |
تستخدم دالة الإعدادات الرئيسية لإنشاء وتحميل النموذج.
|