Update app.py

app.py

@@ -8,8 +8,8 @@ tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 # 設置填充標記 ID
 tokenizer.pad_token = tokenizer.eos_token
 
-def generate_code(prompt, max_length=200):
-    full_prompt = f"Generate Python code for {prompt}:```python\n"
+def generate_command(prompt, max_length=100):
+    full_prompt = f"Generate a command for {prompt}:```bash\n"
     inputs = tokenizer.encode(full_prompt, return_tensors="pt")
     
     output = model.generate(
@@ -21,25 +21,26 @@ def generate_code(prompt, max_length=200):
     )
     generated_text = tokenizer.decode(output[0], skip_special_tokens=False)
     
-    # 提取生成的代碼
-    start = generated_text.find("```python") + len("```python")
+    # 提取生成的指令
+    start = generated_text.find("```bash") + len("```bash")
     end = generated_text.find("```", start)
     if end == -1:
         end = len(generated_text)
-    code = generated_text[start:end].strip()
+    command = generated_text[start:end].strip()
     
-    return code
+    return command
 
 def predict(input_text):
-    output = generate_code(input_text)
+    output = generate_command(input_text)
     return output
 
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your code generation prompt here..."),
+    inputs=gr.Textbox(lines=2, placeholder="Enter your command generation prompt here..."),
     outputs="text",
-    title="Code Generation with GPT2",
-    description="Generate Python code based on your input prompt."
+    title="Command Generation with GPT2",
+    description="Generate bash commands based on your input prompt."
 )
 
-iface.launch()
+iface.launch()
+```