Base files

Dockerfile

@@ -0,0 +1,19 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+# Get secret openai_api_token and output it to /test at buildtime
+RUN --mount=type=secret,id=openai_api_token,mode=0444,required=true \
+   cat /run/secrets/openai_api_token > /test
+
+# Get secret openai_api_token and clone it as repo at buildtime
+RUN --mount=type=secret,id=openai_api_token,mode=0444,required=true \
+  git clone $(cat /run/secrets/openai_api_token)
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -5,6 +5,7 @@ colorFrom: green
 colorTo: green
 sdk: docker
 pinned: false
+app_port: 7860
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.py

@@ -0,0 +1,25 @@
+import openai
+
+openai.api_key = os.environ.get("openai_api_key")
+
+
+_tweet = "I'm a pretty average middle aged guy, got a solid job and had a good family but things got messy and took " \
+         "a turn for the worst. Everyone says you need to hit rock bottom to go back up so here I am!"
+
+
+def prompt(tweet, roast=True):
+    roast_or_toast = "snarky joke" if roast else "encouragement"
+    return f"A user tweeted this tweet:\n\n{tweet}\\n\nThe following {roast_or_toast} was given as an answer:"
+
+
+response = openai.Completion.create(
+    engine="text-davinci-002",
+    prompt=prompt(_tweet, roast=False),
+    temperature=0.7,
+    max_tokens=60,
+    top_p=1,
+    frequency_penalty=0,
+    presence_penalty=0
+)
+
+print(response.choices[0].text.strip())

requirements.txrt

@@ -0,0 +1,5 @@
+fastapi==0.74.*
+requests==2.27.*
+sentencepiece==0.1.*
+uvicorn[standard]==0.17.*
+openai