Update app.py

app.py

@@ -1,8 +1,16 @@
-# Use a pipeline as a high-level helper
+import gradio as gr
 from transformers import pipeline
 
-messages = [
-    {"role": "user", "content": "Who are you?"},
-]
+# 加载模型
 pipe = pipeline("text-generation", model="IndexTeam/Index-1.9B-Character", trust_remote_code=True)
-pipe(messages)
+
+# 定义Gradio接口
+def generate_text(prompt):
+    messages = [
+        {"role": "user", "content": prompt},
+    ]
+    result = pipe(messages)
+    return result[0]['generated_text']
+
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
+iface.launch()