FilesExpand file tree

 main

/

model.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

30 lines (22 loc) · 1020 Bytes

 main

/

model.py

Copy path

Top

File metadata and controls

• Code
• Blame

30 lines (22 loc) · 1020 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

from typing import List

from replicate.base_model import BaseModel

from replicate.collection import Collection

from replicate.exceptions import ReplicateException

from replicate.version import VersionCollection

class Model(BaseModel):

username: str

name: str

def predict(self, *args, **kwargs):

raise ReplicateException(

f"The `model.predict()` method has been removed, because it's unstable: if a new version of the model you're using is pushed and its API has changed, your code may break. Use `version.predict()` instead. See https://github.com/replicate/replicate-python#readme"

)

@property

def versions(self):

return VersionCollection(client=self._client, model=self)

class ModelCollection(Collection):

model = Model

def get(self, name: str) -> Model:

# TODO: fetch model from server

# TODO: support permanent IDs

username, name = name.split("/")

return self.prepare_model({"username": username, "name": name})