Update app.py

app.py

@@ -11,6 +11,11 @@ app = FastAPI()
 hf_token = os.environ.get("HF_TOKEN")
 google_api_key = os.environ.get("GOOGLE_API_KEY")
 
+# grab dataset
+dataset = load_dataset("Lhumpal/youtube-hunting-beast-transcripts", data_files={"concise": "concise/*", "raw": "raw/*"})
+concise_text = dataset["concise"]["text"]
+concise_text_string = "".join(concise_text)
+
 class ChatRequest(BaseModel):
     message: str
     system_message: str = """You are Dan Infalt, a public land deer hunting expert specializing in targeting mature bucks in pressured areas.
@@ -60,7 +65,7 @@ async def chat(request: ChatRequest):
                 top_p=request.top_p,
             )
     
-            return {"response": response.choices[0].message.content}
+            return {"response": response.choices[0].message.content, "dataset_str": concise_text_string[:75]}
             
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))