seed

app.py

@@ -16,14 +16,22 @@ description = "This model has to run on GPU. By default, we load the model with
 
 article = "<p style='text-align: center'><a href='https://news.machinelearning.sg/posts/beautiful_profile_pics_remove_background_image_with_deeplabv3/'>Blog</a> | <a href='https://github.com/eugenesiow/practical-ml'>Github Repo</a></p>"
 
-def MoMA_demo(rgb, subject, prompt, strength):
-    # move the input and model to GPU for speed if available
+def MoMA_demo(rgb, subject, prompt, strength, seed):
+    seed = int(seed) if seed else 0
+    try:
+        seed = int(seed)
+    except ValueError:
+        seed = np.random.randint(0,1000)
+    if seed == 0:
+        seed = np.random.randint(0,1000)
+    print(f"Seed: {seed}")
+    
     with torch.no_grad():
-        generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=np.random.randint(0,1000))
+        generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=seed)
     return generated_image
 
-def inference(rgb, subject, prompt, strength):
-    result = MoMA_demo(rgb, subject, prompt, strength)
+def inference(rgb, subject, prompt, strength, seed):
+    result = MoMA_demo(rgb, subject, prompt, strength, seed)
     return result
 
 seed_everything(0)
@@ -31,23 +39,16 @@ args = parse_args()
 #load MoMA from HuggingFace. Auto download
 model = MoMA_main_modal(args).to(args.device, dtype=torch.float16)
 
-
-################ change texture ##################
-# prompt = "A wooden sculpture of a car on the table."
-# generated_image = model.generate_images(rgb_path, mask_path, subject, prompt, strength=0.4, seed=4, return_mask=True)  # set strength to 0.4 for better prompt fidelity
-# save_image(generated_image,f"{args.output_path}/{subject}_{prompt}.jpg")
-
-
 gr.Interface(
     inference,
     [gr.Image(type="pil", label="Input RGB"),
      gr.Textbox(lines=1, label="subject"),
      gr.Textbox(lines=1, label="Prompt"),
-     gr.Slider(minimum=0.2, maximum=1.2, step=0.1,label="Strength",default=1.0)],
+     gr.Slider(minimum=0.2, maximum=1.2, step=0.1,label="Strength. Recommend: 1.0 for context editing; 0.4 for texture editing",value=1.0),
+     gr.Textbox(lines=1, label="Seed. Use 0 for a random seed")],
     gr.Image(type="pil", label="Output"),
     title=title,
     description=description,
     article=article,
-    examples=[["example_images/newImages/3.jpg",'car','A car in autumn with falling leaves.',1.0],["example_images/newImages/1.jpeg",'bird','A bird in spring with flowers.',1.0]],
-    # enable_queue=True
-).launch(debug=False)
+    examples=[["example_images/newImages/3.jpg",'car','A car in autumn with falling leaves.',1.0,"1"],["example_images/newImages/3.jpg",'car','A wooden sculpture of a car in a living room.',0.3,"4"],["example_images/newImages/2.jpg",'car','A car on a city road with trees and buildings.',1.0,"1"],["example_images/newImages/1.jpeg",'bird','A bird in spring with flowers.',1.0,"1"]],
+).launch(debug=False)

flagged/log.csv

@@ -1,2 +1,4 @@
 Input RGB,subject,Prompt,Strength,Output,flag,username,timestamp
 ,,,0.2,,,,2024-05-21 18:52:10.695138
+/common/users/ks1418/paper_experiments_users/intern_bytedance/demo/MoMA_demo/flagged/Input RGB/077d773e50de2feb1354087dfc5e13d24d644f05/tmpdedk30g1.jpg,car,A car in autumn with falling leaves.,1,1,,,,2024-05-21 19:16:56.004897
+/common/users/ks1418/paper_experiments_users/intern_bytedance/demo/MoMA_demo/flagged/Input RGB/295e9eb571bfb4dae8b04061d57e2f553cbdc88d/tmpg5bekvxe.jpg,bird,A bird in spring with flowers.,1,,/common/users/ks1418/paper_experiments_users/intern_bytedance/demo/MoMA_demo/flagged/Output/9dad60153c76f2169d506f8bf7f29e77cfb01c82/tmps_j3zkvr.png,,,2024-05-21 19:24:12.269082