Upload handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -15,7 +15,7 @@ device = 0 if torch.cuda.is_available() else -1
|
|
| 15 |
# multi-model list
|
| 16 |
multi_model_list = [
|
| 17 |
{"model_id": "MahmoudIbrahim/Mistral_Nemo_Arabic", "task": "text-generation"},
|
| 18 |
-
{"model_id": "
|
| 19 |
]
|
| 20 |
|
| 21 |
class EndpointHandler():
|
|
|
|
| 15 |
# multi-model list
|
| 16 |
multi_model_list = [
|
| 17 |
{"model_id": "MahmoudIbrahim/Mistral_Nemo_Arabic", "task": "text-generation"},
|
| 18 |
+
{"model_id": "Naseej/noon-7b", "task": "text-generation"},
|
| 19 |
]
|
| 20 |
|
| 21 |
class EndpointHandler():
|