Spaces:
Running
on
Zero
Running
on
Zero
Update model/models/fal_api_models.py
Browse files
model/models/fal_api_models.py
CHANGED
|
@@ -54,24 +54,6 @@ class FalModel():
|
|
| 54 |
return result
|
| 55 |
elif self.model_type == "image2image":
|
| 56 |
raise NotImplementedError("image2image model is not implemented yet")
|
| 57 |
-
# assert "image" in kwargs or "image_url" in kwargs, "image or image_url is required for image2image model"
|
| 58 |
-
# if "image" in kwargs:
|
| 59 |
-
# image_url = None
|
| 60 |
-
# pass
|
| 61 |
-
# handler = fal_client.submit(
|
| 62 |
-
# f"fal-ai/{self.model_name}",
|
| 63 |
-
# arguments={
|
| 64 |
-
# "image_url": image_url
|
| 65 |
-
# },
|
| 66 |
-
# )
|
| 67 |
-
#
|
| 68 |
-
# for event in handler.iter_events():
|
| 69 |
-
# if isinstance(event, fal_client.InProgress):
|
| 70 |
-
# print('Request in progress')
|
| 71 |
-
# print(event.logs)
|
| 72 |
-
#
|
| 73 |
-
# result = handler.get()
|
| 74 |
-
# return result
|
| 75 |
elif self.model_type == "text2video":
|
| 76 |
assert "prompt" in kwargs, "prompt is required for text2video model"
|
| 77 |
if self.model_name == 'AnimateDiff':
|
|
|
|
| 54 |
return result
|
| 55 |
elif self.model_type == "image2image":
|
| 56 |
raise NotImplementedError("image2image model is not implemented yet")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
elif self.model_type == "text2video":
|
| 58 |
assert "prompt" in kwargs, "prompt is required for text2video model"
|
| 59 |
if self.model_name == 'AnimateDiff':
|