Update app.py

app.py

@@ -57,6 +57,7 @@ class MultiModalModel(nn.Module):
                 do_sample=True
             )
             return self.code_tokenizer.decode(outputs[0], skip_special_tokens=True)
+        # 添加其他任务的逻辑...
 
 # 定义 Gradio 接口的推理函数
 def gradio_inference(task, input_text):
@@ -79,7 +80,7 @@ model = MultiModalModel()
 # 创建 Gradio 接口
 interface = gr.Interface(
     fn=gradio_inference,
-    inputs=[gr.inputs.Dropdown(choices=["text_generation", "code_generation"], label="任务类型"), gr.inputs.Textbox(lines=2, placeholder="输入文本...")],
+    inputs=[gr.Dropdown(choices=["text_generation", "code_generation"], label="任务类型"), gr.Textbox(lines=2, placeholder="输入文本...")],
     outputs="text",
     title="多模态模型推理",
     description="选择任务类型并输入文本以进行推理"