Omnibus commited on
Commit
edba60c
·
1 Parent(s): 4e83622

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import bs4
5
  import lxml
6
  import os
7
  #import subprocess
8
- from huggingface_hub import InferenceClient,HfApi
9
  import random
10
  import json
11
  import datetime
@@ -21,7 +21,7 @@ from agent import (
21
  LOG_RESPONSE,
22
  )
23
  api=HfApi()
24
-
25
  client = InferenceClient(
26
  "mistralai/Mixtral-8x7B-Instruct-v0.1"
27
  )
@@ -388,6 +388,23 @@ def summarize(inp,history,data=None,files=None,url=None,pdf_url=None,pdf_batch=N
388
  #################################
389
  def clear_fn():
390
  return "",[(None,None)]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
391
  def itt(url):
392
  out=[]
393
  try:
 
5
  import lxml
6
  import os
7
  #import subprocess
8
+ from huggingface_hub import InferenceClient,HfApi,upload_file
9
  import random
10
  import json
11
  import datetime
 
21
  LOG_RESPONSE,
22
  )
23
  api=HfApi()
24
+ tok=os.environ.get("HF_TOKEN")
25
  client = InferenceClient(
26
  "mistralai/Mixtral-8x7B-Instruct-v0.1"
27
  )
 
388
  #################################
389
  def clear_fn():
390
  return "",[(None,None)]
391
+
392
+ def hf_write(inp,f_name="tmp1",repo_name="Omnibus/tmp1",token=tok):
393
+ uidd=uuid.uuid4()
394
+ with open(f"{uidd}.csv", "w") as f:
395
+ f.write(app)
396
+
397
+ api.upload_file(
398
+ path_or_fileobj=f"{uidd}.csv",
399
+ path_in_repo=f"{f_name}.csv",
400
+ repo_id=repo_name,
401
+ token=token,
402
+ repo_type="dataset",
403
+ )
404
+ os.remove(f"{uidd}.csv")
405
+
406
+
407
+
408
  def itt(url):
409
  out=[]
410
  try: