arxivgpt kim commited on
Commit
3cb0117
โ€ข
1 Parent(s): 2d07bd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -9
app.py CHANGED
@@ -80,15 +80,19 @@ def get_caption_from_kosmos(image_in):
80
  return truncated_caption
81
 
82
  def get_caption(image_in):
83
- client = Client("https://fffiloni-moondream1.hf.space/", hf_token=hf_token)
84
- result = client.predict(
85
- image_in, # filepath in 'image' Image component
86
- "Describe precisely the image in one sentence.", # str in 'Question' Textbox component
87
- #api_name="/answer_question"
88
- api_name="/predict"
89
- )
90
- print(result)
91
- return result
 
 
 
 
92
 
93
  def get_magnet(prompt):
94
  amended_prompt = f"{prompt}"
 
80
  return truncated_caption
81
 
82
  def get_caption(image_in):
83
+ try:
84
+ client = Client("https://vikhyatk-moondream1.hf.space/", hf_token=hf_token)
85
+ result = client.predict(
86
+ image_in,
87
+ "Describe precisely the image in one sentence.",
88
+ api_name="/predict"
89
+ )
90
+ print(result)
91
+ return result
92
+ except Exception as e:
93
+ print(f"API ํ˜ธ์ถœ ์ค‘ ์˜ค๋ฅ˜ ๋ฐœ์ƒ: {e}")
94
+ return "API ํ˜ธ์ถœ ์‹คํŒจ๋กœ ์บก์…˜์„ ์ƒ์„ฑํ•  ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค."
95
+
96
 
97
  def get_magnet(prompt):
98
  amended_prompt = f"{prompt}"