data_upload / app.py
Krishnavardhan's picture
Upload app.py
deadd78 verified
raw
history blame
1.3 kB
import gradio as gr
from huggingface_hub import HfApi
import os
# Replace these with your own values
HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
REPO_ID = "Krishnavardhan/data_upload" # Space to upload the files
FOLDER_NAME = "data"
# Hugging Face API instance
api = HfApi()
# Function to upload file to Hugging Face space
def upload_to_huggingface(file):
# Extract only the file name, without the full path
file_name = os.path.basename(file.name)
path_in_repo = f"{FOLDER_NAME}/{file_name}"
try:
# Upload the file to the specified space
api.upload_file(
path_or_fileobj=file.name, # Upload the actual file from the full local path
path_in_repo=path_in_repo, # Use only the file name when uploading to Hugging Face
repo_id=REPO_ID,
repo_type="space",
token=HUGGING_FACE_TOKEN
)
return f"File '{file_name}' successfully uploaded to Hugging Face space!"
except Exception as e:
return f"Error uploading file: {str(e)}"
# Gradio interface for file upload
def file_upload_interface(file):
return upload_to_huggingface(file)
# Launch the Gradio interface
gr.Interface(fn=file_upload_interface, inputs="file", outputs="text").launch()