ariG23498 HF staff commited on
Commit
6419a66
·
verified ·
1 Parent(s): 450fdea

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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.post("/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."},
 
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."},