Update app.py

app.py

@@ -1,26 +1,12 @@
 import gradio as gr
-import requests
+from transformers import pipeline
 
-# 定义内部逻辑函数
-def get_ollama_response(query):
-    try:
-        # 模拟 API 请求（替换为实际逻辑）
-        response = requests.post("http://127.0.0.1:7860/api/query", json={"query": query})
-        if response.status_code != 200:
-            return f"Error: Server returned status code {response.status_code}"
-        
-        response_data = response.json()
-        return response_data.get("answer", "No answer found in response.")
-    except Exception as e:
-        return f"Error: {str(e)}"
+# 加载 CodeGen 模型
+code_generator = pipeline('text-generation', model='Salesforce/codegen-350M-multi')
 
-# 创建 Gradio 界面
-iface = gr.Interface(
-    fn=get_ollama_response,  # 输入输出处理函数
-    inputs="text",           # 输入为文本
-    outputs="text",          # 输出为文本
-    title="Hugging Face Gradio App",
-)
+def generate_code(prompt):
+    return code_generator(prompt, max_length=100)[0]['generated_text']
 
-# 启动 Gradio 应用
-iface.launch(server_name="0.0.0.0", server_port=7860, share=True)
+# 创建 Gradio 接口
+iface = gr.Interface(fn=generate_code, inputs="text", outputs="text")
+iface.launch()