Spaces:

ugonna
/

NaiBot_API

Sleeping

ugonna commited on Jun 5

Commit

3a5bab4

verified ·

1 Parent(s): aa09e62

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -32,7 +32,7 @@ class ModelManager:
         self.device = None
         self.loading_status = "not_started"
         # Use a model from Hugging Face Hub
-        self.model_id = os.environ.get("MODEL_ID", "microsoft/DialoGPT-small")
     def setup_device(self):
         if torch.cuda.is_available():

         self.device = None
         self.loading_status = "not_started"
         # Use a model from Hugging Face Hub
+        self.model_id = os.environ.get("MODEL_ID", "ugonna/llama3.18B-Fine-tunedByUgo3")
     def setup_device(self):
         if torch.cuda.is_available():