pratikshahp commited on
Commit
dcda425
·
verified ·
1 Parent(s): cbc8866

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,12 +24,12 @@ def generate_compliment(image):
24
  image_bytes = buffered.getvalue()
25
 
26
  # Connect to the captioning model on Hugging Face Spaces
27
- captioning_url = "https://gokaygokay-sd3-long-captioner.hf.space/run/create_captions_rich"
28
  caption_response = requests.post(captioning_url, files={"image": image_bytes})
29
  caption = caption_response.json()["data"][0]
30
 
31
  # Connect to the LLM model on Hugging Face Spaces
32
- llm_url = "https://hysts-zephyr-7b.hf.space/run/chat"
33
  llm_payload = {
34
  "system_prompt": SYSTEM_PROMPT,
35
  "message": f"Caption: {caption}\nCompliment: ",
 
24
  image_bytes = buffered.getvalue()
25
 
26
  # Connect to the captioning model on Hugging Face Spaces
27
+ captioning_url = "https://huggingface.co/spaces/gokaygokay/SD3-Long-Captioner"
28
  caption_response = requests.post(captioning_url, files={"image": image_bytes})
29
  caption = caption_response.json()["data"][0]
30
 
31
  # Connect to the LLM model on Hugging Face Spaces
32
+ llm_url = "https://huggingface.co/spaces/hysts/zephyr-7b"
33
  llm_payload = {
34
  "system_prompt": SYSTEM_PROMPT,
35
  "message": f"Caption: {caption}\nCompliment: ",