Skip to content

Instantly share code, notes, and snippets.

@nickstenning

nickstenning/cog.yaml

Created December 6, 2023 19:29
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save nickstenning/d612876d5cce449cd3670caf894a7965 to your computer and use it in GitHub Desktop.
Save nickstenning/d612876d5cce449cd3670caf894a7965 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cog.yaml
build:
gpu: false
python_version: "3.11"
predict: "predict.py:Predictor"
Raw
predict.py
from cog import BasePredictor, Input, Path
class Predictor(BasePredictor):
def setup(self):
print("Doing something very complicated...")
def predict(
self,
name: str = Input(description="Who to greet"),
) -> str:
return f"Hello, {name}!"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment