attempt at a helloworld FastApi endpoint

app.py

@@ -1,16 +1,37 @@
-import streamlit as st
-
+# import streamlit as st
 # x = st.slider('Select a value')
 # st.write(x, 'squared is', x * x)
 
-import torch
+# The code below confirms there is no GPU available
+# import torch
+# st.write("torch version: " + torch.__version__)
+# st.write("cuda available: " + str(torch.cuda.is_available()))
+# st.write("No of GPUs: " + str(torch.cuda.device_count()))
+
+# Hypothesis: Huggingface will host a static copy of a model so we can use that in a FastAPI endpoint to be called externally
+# Requirement: ability to get anything back from a post request of an API hosted on Huiggingface server 
+# (separate to the requirement of a GPU, which will cost money)
+
+from fastapi import FastAPI, Request, HTTPException
+from payload import SomeText
+
+app=FastAPI(title="Hello Huggingworld",
+            version="1.0",
+            debug=True,
+            swagger_ui_bundle_js= "//unpkg.com/swagger-ui-dist@3/swagger-ui-bundle.js",
+            swagger_ui_standalone_preset_js= "//unpkg.com/swagger-ui-dist@3/swagger-ui-standalone-preset.js",
+            summary="API to return the text belonging to a payload."
+)
+
+@app.post("/api/test-io-capability")
+async def test_io(request: Request, input: SomeText): #
+    """
+    Takes in some raw text and returns the same
+    """ 
+
+    if len(input.text) > 0:
 
-print(torch.__version__)
+        return input.text
 
-st.write("torch version: " + torch.__version__)
-st.write(" ")
-st.write("cuda available: " + str(torch.cuda.is_available()))
-st.write(" ")
-st.write("No of GPUs: " + str(torch.cuda.device_count()))
-st.write(" ")
-st.write("device name: " + torch.cuda.get_device_name())
+    else:
+        raise HTTPException(status_code=400, detail = "payload contains no text")

payload.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel
+
+class SomeText(BaseModel):
+    """ This Data Class defines ...
+    """
+    text: str
+    model_config = {
+        "json_schema_extra": {
+            "examples": [
+                {
+                    "text": "twoday is gonna be the day that they're gonna throw it back to you."
+                }
+            ]
+        }
+    }

requirements.txt

@@ -1 +1,3 @@
-torch
+torch
+pydantic
+fastapi