Skip to content
Snippets Groups Projects
Verified Commit 82adb609 authored by Idriss Neumann's avatar Idriss Neumann
Browse files

Update README

parent 4d1f33c8
No related branches found
No related tags found
No related merge requests found
Pipeline #24011 passed
# Comwork AI
Comwork AI API based on GPT-2
Comwork AI API that bring all the AI models you want as an API using a ModelDriver's abstract:
This API is publicly available here: https://cwai-api.comwork.io
```python
class ModelDriver(ABC):
@abstractmethod
def load_model(self):
pass
You have also a frontend chat app available here: https://cwai.comwork.io
@abstractmethod
def generate_response(self, prompt: Prompt):
pass
```
This API is also used as an internal component for Comwork Cloud, more details here: https://doc.cloud.comwork.io/docs/tutorials/cwai
## Git repositories
......@@ -32,6 +41,7 @@ Then you can open http://localhost:8000 and test the API via Swagger.
* GPT2
* Bloom (beware it'll take more than 300Gb of storage)
* bert-base-multilingual-uncased-sentiment
In order to avoid downloading to much data, you can override the list of the models with this variable:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment