Update app.py

app.py

@@ -49,7 +49,7 @@ async def chat(request: ChatRequest):
                     system_instruction=[request.system_message]
                 ),
             )
-            return {"response": response.text}
+            return {"response": response.text, "dataset_str": concise_text_string[:75]}
 
         if request.model_choice == "HF":
             if hf_token:
@@ -69,7 +69,7 @@ async def chat(request: ChatRequest):
                 top_p=request.top_p,
             )
     
-            return {"response": response.choices[0].message.content, "dataset_str": concise_text_string[:75]}
+            return {"response": response.choices[0].message.content}
             
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))