Update main.py

main.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 
 # 初始化pipeline
-pipe_gpt = pipeline("text-generation", model="FreedomIntelligence/Apollo-2B", trust_remote_code=True)
+pipe_gpt = pipeline("text-generation", model="FreedomIntelligence/Apollo-0.5B", trust_remote_code=True)
 
 def generate_text(input_text):
     output = pipe_gpt(input_text, max_length=500)  # 调整max_length来控制生成文本的长度
@@ -12,8 +12,8 @@ def generate_text(input_text):
 iface = gr.Interface(fn=generate_text, 
                      inputs=gr.inputs.Textbox(lines=5, placeholder="Type your prompt here..."), 
                      outputs="text", 
-                     title="Text Generation with Apollo-2B",
-                     description="Enter some text to see how Apollo-2B continues it.")
+                     title="Text Generation with Apollo-0.5B",
+                     description="Enter some text to see how Apollo-0.5B continues it.")
 
 # 运行Gradio应用
 iface.launch()