Spaces:
Paused
Paused
Update main.py
Browse files
main.py
CHANGED
@@ -24,7 +24,7 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
24 |
def read_root():
|
25 |
return {"message": "API is live. Use the /predict endpoint."}
|
26 |
|
27 |
-
@app.
|
28 |
def predict(image_url: str = Query(...), prompt: str = Query(...)):
|
29 |
messages = [
|
30 |
{"role": "system", "content": "You are a helpful assistant with vision abilities."},
|
|
|
24 |
def read_root():
|
25 |
return {"message": "API is live. Use the /predict endpoint."}
|
26 |
|
27 |
+
@app.get("/predict")
|
28 |
def predict(image_url: str = Query(...), prompt: str = Query(...)):
|
29 |
messages = [
|
30 |
{"role": "system", "content": "You are a helpful assistant with vision abilities."},
|