Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,8 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
|
|
12 |
model = AutoModelForCausalLM.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True).eval()#.to("cuda").eval()
|
13 |
processor = AutoProcessor.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True)
|
14 |
|
15 |
-
TITLE = "# [thwri/CogFlorence-2.1-Large]"
|
16 |
-
DESCRIPTION = "microsoft/Florence-2-large tuned on Ejafa/ye-pop captioned with CogVLM2"
|
17 |
|
18 |
def modify_caption(caption: str) -> str:
|
19 |
special_patterns = [
|
|
|
12 |
model = AutoModelForCausalLM.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True).eval()#.to("cuda").eval()
|
13 |
processor = AutoProcessor.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True)
|
14 |
|
15 |
+
TITLE = "# [thwri/CogFlorence-2.1-Large](https://huggingface.co/thwri/CogFlorence-2.1-Large/)"
|
16 |
+
DESCRIPTION = "[microsoft/Florence-2-large](https://huggingface.co/microsoft/Florence-2-large) tuned on [Ejafa/ye-pop](https://huggingface.co/datasets/Ejafa/ye-pop) captioned with [CogVLM2](https://huggingface.co/THUDM/cogvlm2-llama3-chat-19B)"
|
17 |
|
18 |
def modify_caption(caption: str) -> str:
|
19 |
special_patterns = [
|