Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ torch.hub.download_url_to_file(
|
|
29 |
|
30 |
def generate_caption_longcap(image_path: str, return_job: bool = False) -> str | Job:
|
31 |
try:
|
32 |
-
client = Client("
|
33 |
fn = client.submit if return_job else client.predict
|
34 |
return fn(image_path, api_name="/caption")
|
35 |
except Exception:
|
|
|
29 |
|
30 |
def generate_caption_longcap(image_path: str, return_job: bool = False) -> str | Job:
|
31 |
try:
|
32 |
+
client = Client("unography/image-captioning-with-longcap")
|
33 |
fn = client.submit if return_job else client.predict
|
34 |
return fn(image_path, api_name="/caption")
|
35 |
except Exception:
|