Update app.py

app.py

@@ -16,8 +16,11 @@ MODELS = {
 def generate_and_analyze(model_name, input_text):
     model= MODELS[model_name]
     tokenizer = AutoTokenizer.from_pretrained(model)
+    if model == "OpenAssistant":
+        tokenizer = AutoTokenizer.from_pretrained("OpenAssistant")
+        model = AutoModelForCausalLM.from_pretrained("OpenAssistant
+    
     TOKENIZERS[model_name] = tokenizer
-
     text_generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=0) # Use GPU if available
     result = text_generator(input_text, max_length=250, do_sample=True)[0]
     return result['generated_text']
@@ -26,7 +29,7 @@ def generate_and_analyze(model_name, input_text):
 iface = gr.Interface(
     fn=generate_and_analyze, 
     inputs=[
-        gr.inputs.Dropdown(choices=list(MODELS.keys()) + ["bloom"], label="Model"),
+        gr.inputs.Dropdown(choices=list(MODELS.keys()), label="Model"),
         gr.inputs.Textbox(lines=2, label="Input Text")
     ], 
     outputs="text"