first

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/AskMoondream.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,207 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="187">
+            <item index="0" class="java.lang.String" itemvalue="importlib-metadata" />
+            <item index="1" class="java.lang.String" itemvalue="chromadb" />
+            <item index="2" class="java.lang.String" itemvalue="tzlocal" />
+            <item index="3" class="java.lang.String" itemvalue="scikit-learn" />
+            <item index="4" class="java.lang.String" itemvalue="tabulate" />
+            <item index="5" class="java.lang.String" itemvalue="opentelemetry-proto" />
+            <item index="6" class="java.lang.String" itemvalue="validators" />
+            <item index="7" class="java.lang.String" itemvalue="ufw" />
+            <item index="8" class="java.lang.String" itemvalue="pyth" />
+            <item index="9" class="java.lang.String" itemvalue="pycairo" />
+            <item index="10" class="java.lang.String" itemvalue="pycparser" />
+            <item index="11" class="java.lang.String" itemvalue="cmdtest" />
+            <item index="12" class="java.lang.String" itemvalue="PyGObject" />
+            <item index="13" class="java.lang.String" itemvalue="posthog" />
+            <item index="14" class="java.lang.String" itemvalue="distro-info" />
+            <item index="15" class="java.lang.String" itemvalue="pyasn1-modules" />
+            <item index="16" class="java.lang.String" itemvalue="systemd-python" />
+            <item index="17" class="java.lang.String" itemvalue="cliapp" />
+            <item index="18" class="java.lang.String" itemvalue="language-selector" />
+            <item index="19" class="java.lang.String" itemvalue="uvicorn" />
+            <item index="20" class="java.lang.String" itemvalue="pysimdjson" />
+            <item index="21" class="java.lang.String" itemvalue="terminado" />
+            <item index="22" class="java.lang.String" itemvalue="pydantic" />
+            <item index="23" class="java.lang.String" itemvalue="Werkzeug" />
+            <item index="24" class="java.lang.String" itemvalue="pyglet" />
+            <item index="25" class="java.lang.String" itemvalue="streamlit" />
+            <item index="26" class="java.lang.String" itemvalue="opentelemetry-api" />
+            <item index="27" class="java.lang.String" itemvalue="pinecone-client" />
+            <item index="28" class="java.lang.String" itemvalue="reportlab" />
+            <item index="29" class="java.lang.String" itemvalue="PyMuPDFb" />
+            <item index="30" class="java.lang.String" itemvalue="pexpect" />
+            <item index="31" class="java.lang.String" itemvalue="pypdf" />
+            <item index="32" class="java.lang.String" itemvalue="python-apt" />
+            <item index="33" class="java.lang.String" itemvalue="SecretStorage" />
+            <item index="34" class="java.lang.String" itemvalue="ubuntu-advantage-tools" />
+            <item index="35" class="java.lang.String" itemvalue="opentelemetry-util-http" />
+            <item index="36" class="java.lang.String" itemvalue="psutil" />
+            <item index="37" class="java.lang.String" itemvalue="openai" />
+            <item index="38" class="java.lang.String" itemvalue="pyxdg" />
+            <item index="39" class="java.lang.String" itemvalue="nvidia-cublas-cu12" />
+            <item index="40" class="java.lang.String" itemvalue="paramiko" />
+            <item index="41" class="java.lang.String" itemvalue="pydantic_core" />
+            <item index="42" class="java.lang.String" itemvalue="platformdirs" />
+            <item index="43" class="java.lang.String" itemvalue="referencing" />
+            <item index="44" class="java.lang.String" itemvalue="PyJWT" />
+            <item index="45" class="java.lang.String" itemvalue="netifaces" />
+            <item index="46" class="java.lang.String" itemvalue="nvidia-nvjitlink-cu12" />
+            <item index="47" class="java.lang.String" itemvalue="nvidia-cusparse-cu12" />
+            <item index="48" class="java.lang.String" itemvalue="pandocfilters" />
+            <item index="49" class="java.lang.String" itemvalue="py4j" />
+            <item index="50" class="java.lang.String" itemvalue="pyasn1" />
+            <item index="51" class="java.lang.String" itemvalue="jupyter-events" />
+            <item index="52" class="java.lang.String" itemvalue="nvidia-nccl-cu12" />
+            <item index="53" class="java.lang.String" itemvalue="sniffio" />
+            <item index="54" class="java.lang.String" itemvalue="websocket-client" />
+            <item index="55" class="java.lang.String" itemvalue="seaborn" />
+            <item index="56" class="java.lang.String" itemvalue="zipp" />
+            <item index="57" class="java.lang.String" itemvalue="prompt-toolkit" />
+            <item index="58" class="java.lang.String" itemvalue="xgboost" />
+            <item index="59" class="java.lang.String" itemvalue="together" />
+            <item index="60" class="java.lang.String" itemvalue="websockets" />
+            <item index="61" class="java.lang.String" itemvalue="pyarrow" />
+            <item index="62" class="java.lang.String" itemvalue="PyOpenGL" />
+            <item index="63" class="java.lang.String" itemvalue="ubuntu-drivers-common" />
+            <item index="64" class="java.lang.String" itemvalue="PyMuPDF" />
+            <item index="65" class="java.lang.String" itemvalue="scipy" />
+            <item index="66" class="java.lang.String" itemvalue="watchfiles" />
+            <item index="67" class="java.lang.String" itemvalue="tornado" />
+            <item index="68" class="java.lang.String" itemvalue="xkit" />
+            <item index="69" class="java.lang.String" itemvalue="nvidia-cudnn-cu12" />
+            <item index="70" class="java.lang.String" itemvalue="torch" />
+            <item index="71" class="java.lang.String" itemvalue="overrides" />
+            <item index="72" class="java.lang.String" itemvalue="toml" />
+            <item index="73" class="java.lang.String" itemvalue="opentelemetry-semantic-conventions" />
+            <item index="74" class="java.lang.String" itemvalue="pandas" />
+            <item index="75" class="java.lang.String" itemvalue="watchdog" />
+            <item index="76" class="java.lang.String" itemvalue="typing_extensions" />
+            <item index="77" class="java.lang.String" itemvalue="dbus-python" />
+            <item index="78" class="java.lang.String" itemvalue="olefile" />
+            <item index="79" class="java.lang.String" itemvalue="usb-creator" />
+            <item index="80" class="java.lang.String" itemvalue="opentelemetry-instrumentation" />
+            <item index="81" class="java.lang.String" itemvalue="traitlets" />
+            <item index="82" class="java.lang.String" itemvalue="protobuf" />
+            <item index="83" class="java.lang.String" itemvalue="ttystatus" />
+            <item index="84" class="java.lang.String" itemvalue="pygame" />
+            <item index="85" class="java.lang.String" itemvalue="threadpoolctl" />
+            <item index="86" class="java.lang.String" itemvalue="Brlapi" />
+            <item index="87" class="java.lang.String" itemvalue="duplicity" />
+            <item index="88" class="java.lang.String" itemvalue="python-dateutil" />
+            <item index="89" class="java.lang.String" itemvalue="roman" />
+            <item index="90" class="java.lang.String" itemvalue="pycups" />
+            <item index="91" class="java.lang.String" itemvalue="python-dotenv" />
+            <item index="92" class="java.lang.String" itemvalue="gym-maze" />
+            <item index="93" class="java.lang.String" itemvalue="PyPika" />
+            <item index="94" class="java.lang.String" itemvalue="nvidia-cusolver-cu12" />
+            <item index="95" class="java.lang.String" itemvalue="pyzmq" />
+            <item index="96" class="java.lang.String" itemvalue="polars" />
+            <item index="97" class="java.lang.String" itemvalue="pyparsing" />
+            <item index="98" class="java.lang.String" itemvalue="unattended-upgrades" />
+            <item index="99" class="java.lang.String" itemvalue="opentelemetry-instrumentation-asgi" />
+            <item index="100" class="java.lang.String" itemvalue="sympy" />
+            <item index="101" class="java.lang.String" itemvalue="uvloop" />
+            <item index="102" class="java.lang.String" itemvalue="triton" />
+            <item index="103" class="java.lang.String" itemvalue="setuptools-git" />
+            <item index="104" class="java.lang.String" itemvalue="opentelemetry-exporter-otlp-proto-grpc" />
+            <item index="105" class="java.lang.String" itemvalue="pulsar-client" />
+            <item index="106" class="java.lang.String" itemvalue="orjson" />
+            <item index="107" class="java.lang.String" itemvalue="opentelemetry-instrumentation-fastapi" />
+            <item index="108" class="java.lang.String" itemvalue="duckdb" />
+            <item index="109" class="java.lang.String" itemvalue="nvidia-nvtx-cu12" />
+            <item index="110" class="java.lang.String" itemvalue="streamlit-float" />
+            <item index="111" class="java.lang.String" itemvalue="widgetsnbextension" />
+            <item index="112" class="java.lang.String" itemvalue="gyp" />
+            <item index="113" class="java.lang.String" itemvalue="pyautogen" />
+            <item index="114" class="java.lang.String" itemvalue="webargs" />
+            <item index="115" class="java.lang.String" itemvalue="whatsapp-python" />
+            <item index="116" class="java.lang.String" itemvalue="miniworld" />
+            <item index="117" class="java.lang.String" itemvalue="defer" />
+            <item index="118" class="java.lang.String" itemvalue="pydeck" />
+            <item index="119" class="java.lang.String" itemvalue="ptyprocess" />
+            <item index="120" class="java.lang.String" itemvalue="SQLAlchemy" />
+            <item index="121" class="java.lang.String" itemvalue="opentelemetry-sdk" />
+            <item index="122" class="java.lang.String" itemvalue="pyspark" />
+            <item index="123" class="java.lang.String" itemvalue="wcwidth" />
+            <item index="124" class="java.lang.String" itemvalue="types-python-dateutil" />
+            <item index="125" class="java.lang.String" itemvalue="python-debian" />
+            <item index="126" class="java.lang.String" itemvalue="cupshelpers" />
+            <item index="127" class="java.lang.String" itemvalue="onnxruntime" />
+            <item index="128" class="java.lang.String" itemvalue="chroma-hnswlib" />
+            <item index="129" class="java.lang.String" itemvalue="rpds-py" />
+            <item index="130" class="java.lang.String" itemvalue="tomli" />
+            <item index="131" class="java.lang.String" itemvalue="louis" />
+            <item index="132" class="java.lang.String" itemvalue="urllib3" />
+            <item index="133" class="java.lang.String" itemvalue="faiss-gpu" />
+            <item index="134" class="java.lang.String" itemvalue="sseclient-py" />
+            <item index="135" class="java.lang.String" itemvalue="typer" />
+            <item index="136" class="java.lang.String" itemvalue="pandasai" />
+            <item index="137" class="java.lang.String" itemvalue="parso" />
+            <item index="138" class="java.lang.String" itemvalue="prometheus-client" />
+            <item index="139" class="java.lang.String" itemvalue="pyth3" />
+            <item index="140" class="java.lang.String" itemvalue="pillow" />
+            <item index="141" class="java.lang.String" itemvalue="xdg" />
+            <item index="142" class="java.lang.String" itemvalue="command-not-found" />
+            <item index="143" class="java.lang.String" itemvalue="opentelemetry-exporter-otlp-proto-common" />
+            <item index="144" class="java.lang.String" itemvalue="accelerate" />
+            <item index="145" class="java.lang.String" itemvalue="diffusers" />
+            <item index="146" class="java.lang.String" itemvalue="gradio" />
+            <item index="147" class="java.lang.String" itemvalue="huggingface-hub" />
+            <item index="148" class="java.lang.String" itemvalue="PyYAML" />
+            <item index="149" class="java.lang.String" itemvalue="MarkupSafe" />
+            <item index="150" class="java.lang.String" itemvalue="requests" />
+            <item index="151" class="java.lang.String" itemvalue="Jinja2" />
+            <item index="152" class="java.lang.String" itemvalue="markdown-it-py" />
+            <item index="153" class="java.lang.String" itemvalue="fsspec" />
+            <item index="154" class="java.lang.String" itemvalue="jsonschema-specifications" />
+            <item index="155" class="java.lang.String" itemvalue="semantic-version" />
+            <item index="156" class="java.lang.String" itemvalue="tomlkit" />
+            <item index="157" class="java.lang.String" itemvalue="filelock" />
+            <item index="158" class="java.lang.String" itemvalue="gradio_client" />
+            <item index="159" class="java.lang.String" itemvalue="Pygments" />
+            <item index="160" class="java.lang.String" itemvalue="safetensors" />
+            <item index="161" class="java.lang.String" itemvalue="starlette" />
+            <item index="162" class="java.lang.String" itemvalue="mdurl" />
+            <item index="163" class="java.lang.String" itemvalue="jsonschema" />
+            <item index="164" class="java.lang.String" itemvalue="tokenizers" />
+            <item index="165" class="java.lang.String" itemvalue="transformers" />
+            <item index="166" class="java.lang.String" itemvalue="importlib_resources" />
+            <item index="167" class="java.lang.String" itemvalue="aiofiles" />
+            <item index="168" class="java.lang.String" itemvalue="rich" />
+            <item index="169" class="java.lang.String" itemvalue="pydub" />
+            <item index="170" class="java.lang.String" itemvalue="python-multipart" />
+            <item index="171" class="java.lang.String" itemvalue="click" />
+            <item index="172" class="java.lang.String" itemvalue="altair" />
+            <item index="173" class="java.lang.String" itemvalue="attrs" />
+            <item index="174" class="java.lang.String" itemvalue="fastapi" />
+            <item index="175" class="java.lang.String" itemvalue="regex" />
+            <item index="176" class="java.lang.String" itemvalue="toolz" />
+            <item index="177" class="java.lang.String" itemvalue="ruff" />
+            <item index="178" class="java.lang.String" itemvalue="charset-normalizer" />
+            <item index="179" class="java.lang.String" itemvalue="ffmpy" />
+            <item index="180" class="java.lang.String" itemvalue="shellingham" />
+            <item index="181" class="java.lang.String" itemvalue="langchain" />
+            <item index="182" class="java.lang.String" itemvalue="sentence-transformers" />
+            <item index="183" class="java.lang.String" itemvalue="Pillow" />
+            <item index="184" class="java.lang.String" itemvalue="langchain-text-splitters" />
+            <item index="185" class="java.lang.String" itemvalue="langchain_core" />
+            <item index="186" class="java.lang.String" itemvalue="langchain_community" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+    <inspection_tool class="PyUnresolvedReferencesInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredIdentifiers">
+        <list>
+          <option value="spaces" />
+        </list>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.12" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/AskMoondream.iml" filepath="$PROJECT_DIR$/.idea/AskMoondream.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

app.py

@@ -0,0 +1,59 @@
+import spaces
+import torch
+import re
+import gradio as gr
+from threading import Thread
+from transformers import TextIteratorStreamer, AutoTokenizer, AutoModelForCausalLM
+
+import subprocess
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
+model_id = "vikhyatk/moondream2"
+revision = "2024-04-02"
+tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
+moondream = AutoModelForCausalLM.from_pretrained(
+    model_id, trust_remote_code=True, revision=revision,
+    torch_dtype=torch.bfloat16, device_map={"": "cuda"},
+    attn_implementation="flash_attention_2"
+)
+moondream.eval()
+
+
+@spaces.GPU(duration=10)
+def answer_question(img, prompt):
+    image_embeds = moondream.encode_image(img)
+    streamer = TextIteratorStreamer(tokenizer, skip_special_tokens=True)
+    thread = Thread(
+        target=moondream.answer_question,
+        kwargs={
+            "image_embeds": image_embeds,
+            "question": prompt,
+            "tokenizer": tokenizer,
+            "streamer": streamer,
+        },
+    )
+    thread.start()
+
+    buffer = ""
+    for new_text in streamer:
+        buffer += new_text
+        yield buffer.strip()
+
+
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+        # 🌔 moondream2
+        A tiny vision language model. [GitHub](https://github.com/vikhyat/moondream)
+        """
+    )
+    with gr.Row():
+        prompt = gr.Textbox(label="Input", value="Describe this image.", scale=4)
+        submit = gr.Button("Submit")
+    with gr.Row():
+        img = gr.Image(type="pil", label="Upload an Image")
+        output = gr.TextArea(label="Response")
+    submit.click(answer_question, [img, prompt], output)
+    prompt.submit(answer_question, [img, prompt], output)
+
+demo.queue().launch()

requirements.txt

@@ -0,0 +1,4 @@
+timm==0.9.12
+transformers==4.36.2
+einops==0.7.0
+accelerate==0.25.0