Update app.py

app.py

@@ -29,11 +29,10 @@ class MultiModalModel(nn.Module):
         self.nlp_tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased')
         self.speech_processor = AutoTokenizer.from_pretrained('facebook/wav2vec2-base-960h')
         self.vision_processor = AutoTokenizer.from_pretrained('openai/clip-vit-base-patch32')
-    
+
     def forward(self, task, inputs):
         if task == 'text_generation':
             attention_mask = inputs.get('attention_mask')
-            print("输入数据:", inputs)
             outputs = self.text_generator.generate(
                 inputs['input_ids'], 
                 max_new_tokens=100,  
@@ -44,7 +43,6 @@ class MultiModalModel(nn.Module):
                 temperature=0.8,  
                 do_sample=True
             )
-            print("生成的输出:", outputs)
             return self.text_tokenizer.decode(outputs[0], skip_special_tokens=True)
         elif task == 'code_generation':
             attention_mask = inputs.get('attention_mask')
@@ -59,7 +57,6 @@ 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):
@@ -90,4 +87,3 @@ interface = gr.Interface(
 
 # 启动 Gradio 应用
 interface.launch()
-