更新

app.py

@@ -17,26 +17,35 @@ def main():
     input_api_key = gr.inputs.Textbox(label="ChatGPT API Key", lines=1)
     input_api_base = gr.inputs.Textbox(label="ChatGPT API 地址（默认无地址）", lines=1)
     input_url = gr.inputs.Textbox(label="URL", lines=1)
-    gradio_interface = gr.Interface(fn=my_inference_function, inputs=[input_checkbox, input_api_key, input_api_base, input_url], outputs="text")
-    gradio_interface.launch()
 
-def my_inference_function(enabled, api_key, api_base, url):
-    if enabled:
-      os.environ["OPENAI_API_KEY"] = api_key
-      os.environ['OPENAI_API_BASE'] = api_base
-      llm=OpenAI(temperature=0.7, model_name="gpt-3.5-turbo", max_tokens=1024)
-    else: 
-      llm = HuggingFaceHub(repo_id="declare-lab/flan-alpaca-large", model_kwargs={"temperature":0.1, "max_length":512})
-    loader = UnstructuredURLLoader(urls=[url])
-    data = loader.load()
-    chain = load_qa_chain(llm=llm, chain_type="stuff")
-    response = chain.run(input_documents=data, question="""请用中文总结文章的内容，并以下面模版给出结果：
+    def my_inference_function(enabled, api_key, api_base, url):
+        if enabled:
+            os.environ["OPENAI_API_KEY"] = api_key
+            os.environ['OPENAI_API_BASE'] = api_base
+            llm = OpenAI(temperature=0.7, model_name="gpt-3.5-turbo", max_tokens=1024)
+        else:
+            llm = HuggingFaceHub(repo_id="declare-lab/flan-alpaca-large", model_kwargs={"temperature":0.1, "max_length":512})
+        loader = UnstructuredURLLoader(urls=[url])
+        data = loader.load()
+        chain = load_qa_chain(llm=llm, chain_type="stuff")
+        response = chain.run(input_documents=data, question="""请用中文总结文章的内容，并以下面模版给出结果：
 《文章标题》摘要如下：
 ## 一句话描述
 文章摘要内容
 ## 文章略读
 文章要点""")
-    return response
+        return response
+
+    inputs = [input_checkbox]
+    outputs = "text"
+    if input_checkbox.default is False:
+        inputs += [input_url]
+    else:
+        inputs += [input_api_key, input_api_base]
+
+    gradio_interface = gr.Interface(fn=my_inference_function, inputs=inputs, outputs=outputs)
+    gradio_interface.launch()
+
 
 if __name__ == '__main__':
-    main()
+    main()