Update app.py

app.py

@@ -1,6 +1,6 @@
 import json
 import gradio as gr
-from transformers import TransformersMeta
+from transformers import Tool, push_to_hub
 
 def generate_files(title="Text Generation Tool", emoji="🌖", colorFrom="blue", colorTo="blue",
                    sdk="gradio", sdk_version="4.3.0", app_file="app.py", pinned=False,
@@ -32,7 +32,6 @@ tags:
     # Generate app.py content
     app_py_content = '''from transformers.tools.base import launch_gradio_demo
 from {} import {}Tool
-
 launch_gradio_demo({}Tool)
 '''.format(tool_name, tool_name.capitalize(), tool_name.capitalize())
 
@@ -46,17 +45,13 @@ torch
     # Generate text_generator.py content
     text_generator_py_content = '''import os
 from transformers import pipeline
-from transformers import Tool
-
 class {}Tool(Tool):
     name = "{}"
     description = (
         "{}"
     )
-
     inputs = ["text"]
     outputs = ["text"]
-
     def __call__(self, prompt: str):
         token = os.environ['hf']
         text_generator = pipeline(model="microsoft/Orca-2-13b", token=token)
@@ -81,22 +76,21 @@ class {}Tool(Tool):
     with open("{}.py".format(tool_name), "w") as text_generator_py_file:
         text_generator_py_file.write(text_generator_py_content)
 
-    # Create a new space on Hugging Face
-    space = TransformersMeta.create_space(name=tool_name, description=tool_description)
-    
-    # Upload the files to the new space
-    space.upload_file("README.md")
-    space.upload_file("tool_config.json")
-    space.upload_file("app.py")
-    space.upload_file("requirements.txt")
-    space.upload_file("{}.py".format(tool_name))
+    # Upload the files to the Hugging Face Hub
+    push_to_hub(
+        model_id=tool_name,
+        repo_path="./",
+        commit_message="Upload {} tool".format(tool_name),
+        organization="your_organization",  # Replace 'your_organization' with your actual organization
+        private=False  # Set to True if you want the repository to be private
+    )
 
     # Return the generated files for download
     return "README.md", "tool_config.json", "app.py", "requirements.txt", "{}.py".format(tool_name)
 
 
 # Define the inputs for the Gradio interface
-io = gr.Interface(generate_files, 
+io = gr.Interface(generate_files,
                   inputs=["text", "text", "text", "text", "text", "text", "text", "text", "checkbox", "text", "text"],
                   outputs=["text", "text", "text", "text", "text"])