Update app.py
Browse files
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="
|
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
|