Update app.py

app.py

@@ -1,13 +1,28 @@
 import gradio as gr
 from gpt4all import GPT4All
+import urllib
+import json
 
-model = GPT4All("orca-mini-3b-gguf2-q4_0.gguf")
+url = "https://raw.githubusercontent.com/nomic-ai/gpt4all/main/gpt4all-chat/metadata/models3.json"
+response = urlopen(url)
+data_json = json.loads(response.read())
 
-def generate_text(input_text):
+def model_choices():
+    model_list = [data_json[i]['filename'] for i in range(len(data_json))]
+    return model_list
+
+def generate_text(input_text, selected_model):
+    model = GPT4All(selected_model)
     output = model.generate(input_text, max_tokens=100)
     return output
 
+model_dropdown = gr.Dropdown(choices=model_choices(), multiselect=False, label="LLMs to choose from", type="value")
 input_text = gr.Textbox(lines=5, label="Input Text")
 output_text = gr.Textbox(lines=5, label="Generated Text")
 
-gr.Interface(fn=generate_text, inputs=input_text, outputs=output_text, theme = gr.themes.Soft(), analytics_enabled=True, title="GPT4All_experiment").launch()
+gr.Interface(fn=generate_text, 
+             inputs=[input_text, model_dropdown], 
+             outputs=output_text, 
+             theme = gr.themes.Soft(), 
+             analytics_enabled=True, 
+             title="GPT4All Text Generation Experiment").launch()