.

gui.py

@@ -1,8 +1,6 @@
-import spaces
 import os
 import torch
 import logging
-import random
 import gradio as gr
 import diffusers
 from models.upscaler import upscaler_dict_gui
@@ -25,11 +23,12 @@ class GuiSD:
         self.model = None
         print("Loading model...")
         self.model = Model_Diffusers(
-            base_model_id="misri/newrealityxlAllInOne_Newreality40/blob/main/newrealityxlAllInOne_Newreality40.safetensors",
+            base_model_id="misri/newrealityxlAllInOne_Newreality40",  # Corrected repo_id format
             task_name="txt2img",
             vae_model="vaes/sdXL_v10VAEFix.safetensors",
             type_model_precision=torch.float16,
             retain_task_model_in_cache=False,
+            hf_token=hf_token  # Ensure the token is passed
         )
         self.model_list = model_list
         self.task_stablepy = task_stablepy