Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ def generate_and_analyze(model_name, input_text):
|
|
18 |
tokenizer = AutoTokenizer.from_pretrained(model)
|
19 |
if model == "OpenAssistant":
|
20 |
tokenizer = AutoTokenizer.from_pretrained("OpenAssistant")
|
21 |
-
model = AutoModelForCausalLM.from_pretrained("OpenAssistant
|
22 |
|
23 |
TOKENIZERS[model_name] = tokenizer
|
24 |
text_generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=0) # Use GPU if available
|
|
|
18 |
tokenizer = AutoTokenizer.from_pretrained(model)
|
19 |
if model == "OpenAssistant":
|
20 |
tokenizer = AutoTokenizer.from_pretrained("OpenAssistant")
|
21 |
+
model = AutoModelForCausalLM.from_pretrained("OpenAssistant")
|
22 |
|
23 |
TOKENIZERS[model_name] = tokenizer
|
24 |
text_generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=0) # Use GPU if available
|