strength

app.py

@@ -12,17 +12,18 @@ import os
 os.environ["CUDA_VISIBLE_DEVICES"]="0"
 
 title = "MoMA"
-description = "This model has to run on GPU"
+description = "This model has to run on GPU. By default, we load the model with 4-bit quantization to make it fit in smaller hardwares."
+
 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):
+def MoMA_demo(rgb, subject, prompt, strength):
     # move the input and model to GPU for speed if available
     with torch.no_grad():
-        generated_image = model.generate_images(rgb, subject, prompt, strength=1.0, seed=np.random.randint(0,1000))
+        generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=np.random.randint(0,1000))
     return generated_image
 
-def inference(rgb, subject, prompt):
-    result = MoMA_demo(rgb, subject, prompt)
+def inference(rgb, subject, prompt, strength):
+    result = MoMA_demo(rgb, subject, prompt, strength)
     return result
 
 seed_everything(0)
@@ -41,11 +42,12 @@ gr.Interface(
     inference,
     [gr.Image(type="pil", label="Input RGB"),
      gr.Textbox(lines=1, label="subject"),
-     gr.Textbox(lines=5, label="Prompt")],
+     gr.Textbox(lines=1, label="Prompt"),
+     gr.Slider(minimum=0.2, maximum=1.2, step=0.1,label="Strength",default=1.0)],
     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.']],
+    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)

checkpoints/attn_adapters_projectors.th

@@ -0,0 +1 @@
+../../../../../../../home/ks1418/.cache/huggingface/hub/models--KunpengSong--MoMA_llava_7b/blobs/0b432a39e46f01cd9cdb4794b8ef13b9bb0aff2ad6da6800d67fd2ca4af21fa6

flagged/log.csv

@@ -0,0 +1,2 @@
+Input RGB,subject,Prompt,Strength,Output,flag,username,timestamp
+,,,0.2,,,,2024-05-21 18:52:10.695138