Update app.py

app.py

@@ -159,14 +159,14 @@ pipe = load_and_prepare_model()
 
 # text models
 checkpoint = "microsoft/Phi-3.5-mini-instruct"
-captioner = pipeline(model="ydshieh/vit-gpt2-coco-en",device='cuda', task="image-to-text")
-captioner_2 = pipeline(model="Salesforce/blip-image-captioning-base",device='cuda', task="image-to-text")
+#captioner = pipeline(model="ydshieh/vit-gpt2-coco-en",device='cuda', task="image-to-text")
+#captioner_2 = pipeline(model="Salesforce/blip-image-captioning-base",device='cuda', task="image-to-text")
 captioner_3 = pipeline(model="ford442/blip-image-to-text-large-bf16",device='cuda', task="image-to-text")
 model5 = Blip2ForConditionalGeneration.from_pretrained("ford442/blip2-image-to-text-bf16").to('cuda')
 processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16")
-txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
-txt_tokenizer.tokenizer_legacy=False
-model = AutoModelForCausalLM.from_pretrained(checkpoint).to('cuda')
+#txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
+#txt_tokenizer.tokenizer_legacy=False
+#model = AutoModelForCausalLM.from_pretrained(checkpoint).to('cuda')
 
 ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
 text_encoder=CLIPTextModel.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder',token=True).to(device=device, dtype=torch.bfloat16)
@@ -262,7 +262,7 @@ def captioning(img):
     output_prompt.append(response_text)
     print(output_prompt)
     return output_prompt
-
+'''
 def expand_prompt(prompt):
         system_prompt_rewrite = (
             "You are an AI assistant that rewrites image prompts to be more descriptive and detailed."
@@ -315,7 +315,7 @@ def expand_prompt(prompt):
         print(enhanced_prompt_2)
         enh_prompt=[enhanced_prompt,enhanced_prompt_2]
         return enh_prompt 
-    
+'''   
 @spaces.GPU(duration=40)
 def generate_30(
     prompt: str = "",
@@ -396,9 +396,9 @@ def generate_30(
         print(caption)
         print("-- generating further caption --")
 
-        expand_prompt(prompt)
-        expand_prompt(caption)
-        expand_prompt(caption_2)
+        #expand_prompt(prompt)
+        #expand_prompt(caption)
+        #expand_prompt(caption_2)
 
       
         print('-- generating image --')