mobenta commited on
Commit
888d97e
·
verified ·
1 Parent(s): b59906e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base
24
  florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True)
25
 
26
  # Prompt Enhancer
27
- enhancer_long = pipeline("summarization", model="gokaygokay/Lamini-Prompt-Enchance-Long", device=device)
28
 
29
  MAX_SEED = np.iinfo(np.int32).max
30
  MAX_IMAGE_SIZE = 2048
 
24
  florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True)
25
 
26
  # Prompt Enhancer
27
+ enhancer_long = pipeline("summarization", model="mobenta/M_Prompter", device=device)
28
 
29
  MAX_SEED = np.iinfo(np.int32).max
30
  MAX_IMAGE_SIZE = 2048