FilesExpand file tree

 initial-implementation

/

model.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

30 lines (23 loc) · 889 Bytes

 initial-implementation

/

model.py

Copy path

Top

File metadata and controls

• Code
• Blame

30 lines (23 loc) · 889 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 Version

class Model(BaseModel):

username: str

name: str

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

versions = self._client.versions.list(self)

if not versions:

raise ReplicateException(

"No versions found for model %s/%s" % (self.username, self.name)

)

latest_version = versions[0]

return latest_version.predict(*args, **kwargs)

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})