fffiloni commited on
Commit
33b0931
·
verified ·
1 Parent(s): 525de55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,12 +1,15 @@
1
  import gradio as gr
2
  from gradio_client import Client
 
 
 
3
 
4
  def get_caption(image_in):
5
- client = Client("https://vikhyatk-moondream1.hf.space/")
6
  result = client.predict(
7
  image_in, # filepath in 'image' Image component
8
  "Describe the image", # str in 'Question' Textbox component
9
- api_name="/answer_question"
10
  )
11
  print(result)
12
  return result
 
1
  import gradio as gr
2
  from gradio_client import Client
3
+ import os
4
+
5
+ hf_token = os.environ.get("HF_TKN")
6
 
7
  def get_caption(image_in):
8
+ client = Client("https://fffiloni-moondream1.hf.space/", hf_token=hf_token)
9
  result = client.predict(
10
  image_in, # filepath in 'image' Image component
11
  "Describe the image", # str in 'Question' Textbox component
12
+ api_name="/predict"
13
  )
14
  print(result)
15
  return result