Update app.py

app.py

@@ -1,6 +1,5 @@
 from transformers import AutoModel, AutoTokenizer
 import gradio as gr
-import os
 
 # 加载模型和分词器
 model_id = "BioMistral/BioMistral-7B"
@@ -12,8 +11,9 @@ def generate_text(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")
     # 生成文本（这里使用模型的输出作为示例）
     outputs = model(**inputs)
-    # 返回生成的结果（这里只返回最后一个隐藏状态作为示例）
-    return outputs.last_hidden_state
+    
+    # 处理输出（这里返回最后一个隐藏状态的形状作为示例）
+    return outputs.last_hidden_state.tolist()  # 转换为列表以便Gradio显示
 
 # 创建Gradio界面
 iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")