update app.py

app.py

@@ -41,7 +41,7 @@ import cv2
 import numpy as np
 import matplotlib.pyplot as plt
 from lama_cleaner.model_manager import ModelManager
-from lama_cleaner.schema import Config
+from lama_cleaner.schema import Config as lama_Config
 
 # segment anything
 from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator
@@ -280,7 +280,7 @@ def lama_cleaner_process(image, mask):
     else:
         size_limit = int(size_limit)
 
-    config = Config(
+    config = lama_Config(
         ldm_steps=25,
         ldm_sampler='plms',
         zits_wireframe=True,
@@ -324,14 +324,14 @@ def lama_cleaner_process(image, mask):
 # relate anything
 from ram_utils import iou, sort_and_deduplicate, relation_classes, MLP, show_anns, show_mask
 from ram_train_eval import RamModel,RamPredictor
-from mmengine.config import Config
+from mmengine.config import Config as mmengine_Config
 input_size = 512
 hidden_size = 256
 num_classes = 56
 
 # load ram model
 model_path = "./checkpoints/ram_epoch12.pth"
-config = dict(
+ram_config = dict(
     model=dict(
         pretrained_model_name_or_path='bert-base-uncased',
         load_pretrained_weights=False,
@@ -345,7 +345,7 @@ config = dict(
     ),
     load_from=model_path,
 )
-config = Config(config)
+ram_config = mmengine_Config(ram_config)
 
 class Ram_Predictor(RamPredictor):
     def __init__(self, config, device='cpu'):
@@ -358,7 +358,8 @@ class Ram_Predictor(RamPredictor):
         if self.config.load_from is not None:
             self.model.load_state_dict(torch.load(self.config.load_from, map_location=self.device))
         self.model.train()
-ram_model = Ram_Predictor(config, device)
+
+ram_model = Ram_Predictor(ram_config, device)
 
 # visualization
 def draw_selected_mask(mask, draw):
@@ -736,8 +737,8 @@ if __name__ == "__main__":
         mask_source_radio.change(fn=change_radio_display, inputs=[task_type, mask_source_radio], outputs=[text_prompt, inpaint_prompt, mask_source_radio, num_relation])
 
         DESCRIPTION = '### This demo from [Grounded-Segment-Anything](https://github.com/IDEA-Research/Grounded-Segment-Anything). <br>'
-        DESCRIPTION += '### RAM from [RelateAnything](https://github.com/Luodian/RelateAnything). <br>'
-        DESCRIPTION += '### Thanks for their excellent work.'
+        DESCRIPTION += 'RAM from [RelateAnything](https://github.com/Luodian/RelateAnything). <br>'
+        DESCRIPTION += 'Thanks for their excellent work.'
         DESCRIPTION += f'<p>For faster inference without waiting in queue, you may duplicate the space and upgrade to GPU in settings. <a href="https://huggingface.co/spaces/yizhangliu/Grounded-Segment-Anything?duplicate=true"><img style="display: inline; margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space" /></a></p>'
         gr.Markdown(DESCRIPTION)