Upload folder using huggingface_hub

app.py

@@ -173,7 +173,7 @@ def generate_text(data: Generate):
         influencers = re.findall(pattern, response_text)
         influencers = list(set(influencers))
         id_df = pd.DataFrame(influencers, columns=['influencer_uid'])
-        result = id_df.merge(df, on='influencer_uid', how='left').to_dict(orient='records')
+        result = id_df.merge(df, on='influencer_uid', how='left')[['influencer_uid', 'name']].to_dict(orient='records')
         return {"status": 200, "response": re.sub(pattern, '', response_text), "influencers": result }
     else:
         return response.json()