Update app.py

app.py

@@ -12,18 +12,18 @@ def generate_files(title="Text Generation Tool", tool_description="This is a too
     
     # Generate readme content
     readme_content = '''
-    ---
-    title: {}
-    emoji: 🌖
-    colorFrom: blue
-    colorTo: blue
-    sdk: gradio
-    sdk_version: 4.3.0
-    app_file: app.py
-    pinned: false
-    tags:
-      - tool
-    ---
+---
+title: {}
+emoji: 🌖
+colorFrom: blue
+colorTo: blue
+sdk: gradio
+sdk_version: 4.3.0
+app_file: app.py
+pinned: false
+tags:
+  - tool
+---
     '''.format(title)
 
     tool_name = title.replace(" ", "_").lower()
@@ -40,36 +40,39 @@ def generate_files(title="Text Generation Tool", tool_description="This is a too
     tool_config_json = json.dumps(tool_config, indent=4)
 
     # Generate app.py content
-    app_py_content = '''from transformers.tools.base 
-    from transformers import Tool
-    import launch_gradio_demo
-    from {} import {}
-    launch_gradio_demo({}Tool)
+    app_py_content = '''
+from transformers.tools.base 
+from transformers import Tool
+import launch_gradio_demo
+from {} import {}
+launch_gradio_demo({}Tool)
     '''.format( tool_name, tool_class, tool_class)
 
     # Generate requirements.txt content
-    requirements_content = '''transformers>=4.29.0
-    # diffusers
-    accelerate
-    torch
+    requirements_content = '''
+transformers>=4.29.0
+# diffusers
+accelerate
+torch
     '''
 
     # Generate text_generator.py content
-    text_generator_py_content = '''import os
+    text_generator_py_content = '''
+import os
 from transformers import pipeline
 class {}(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)
-        generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)
-        print(generated_text)
-        return generated_text
+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)
+    generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)
+    print(generated_text)
+    return generated_text
 '''.format(tool_class, tool_name, tool_description)
 
     # Create a new folder for the tool
@@ -112,7 +115,7 @@ io = gr.Interface(generate_files,
                                 label="Titel",
                                 info="Initial text",
                                 lines=1,
-                                value="The quick brown fox jumped over the lazy dogs.",
+                                value="Cool Tool3",
                             ),
                             gr.Textbox(
                                 label="Text 2",