update'

Dockerfile

@@ -18,6 +18,7 @@ RUN pip install --no-cache-dir --upgrade pip && \
         torchvision \
         git+https://github.com/huggingface/transformers \
         accelerate \
+        transformers \
         qwen-vl-utils[decord]==0.0.8 \
         fastapi \
         uvicorn[standard] 

README.md

@@ -8,6 +8,12 @@ pinned: false
 license: apache-2.0
 ---
 
+curl -X GET "http://localhost:8000/predict?image_url=https://example.com/sample-image.jpg&prompt=Describe%20the%20contents%20of%20this%20image." -H "accept: application/json"
+
+curl -X GET "http://localhost:8000/chat?prompt=Hello%20there%2C%20how%20are%20you%3F" -H "accept: application/json"
+
+
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 

main.py.old

@@ -36,6 +36,7 @@ def read_root():
 
 @app.get("/predict")
 def predict(image_url: str = Query(...), prompt: str = Query(...)):
+    
     messages = [
         {"role": "system", "content": "You are a helpful assistant with vision abilities."},
         {"role": "user", "content": [{"type": "image", "image": image_url}, {"type": "text", "text": prompt}]},