Update app.py

app.py

@@ -11,17 +11,14 @@ app = FastAPI()
 hf_token = os.environ.get("HF_TOKEN")
 # login(token=hf_token)
 
-# client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
 # dataset = load_dataset("Lhumpal/youtube-hunting-beast-transcripts", data_files={"concise": "concise/*", "raw": "raw/*"})
-dataset = None
-
-if dataset:
-    texts = []
-    for file in dataset["concise"]:
-        # Remove newline characters from the 'text' field
-        cleaned_text = file['text'].replace('\n', ' ')
-        texts.append(cleaned_text)
+# if dataset:
+#     texts = []
+#     for file in dataset["concise"]:
+#         # Remove newline characters from the 'text' field
+#         cleaned_text = file['text'].replace('\n', ' ')
+#         texts.append(cleaned_text)
 
 class ChatRequest(BaseModel):
     message: str
@@ -56,7 +53,7 @@ async def chat(request: ChatRequest):
             token = message.choices[0].delta.content
             response += token
 
-        return {"assistant_response": response, "dataset_sample": "sample"}
+        return {"assistant_response": response}
         
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))