app

app.py

@@ -8,7 +8,7 @@ from PIL import ImageDraw
 from torchvision.transforms.v2 import Resize
 import subprocess
 
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+#subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 model_id = "vikhyatk/moondream2"
 revision = "2025-01-09"
@@ -65,4 +65,4 @@ with gr.Blocks() as demo:
         output2 = gr.Dataframe(label="Structured Dataframe", type="array", wrap=True)
     submit.click(answer_questions, inputs=[img, prompt], outputs=[output, output2])
 
-demo.queue().launch()
+demo.queue().launch()