Spaces:

acecalisto3
/

CEEMEESEEK

Runtime error

acecalisto3 commited on Oct 8, 2024

Commit

2bfb5cf

verified ·

1 Parent(s): 280b21a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False, legacy=Fal
 model = OpenLlamaForCausalLM.from_pretrained(model_name)
 openllama_pipeline = pipeline(
-    "text-generation",
     model=model,
     tokenizer=tokenizer,
     truncation=True,  # Add this line

 model = OpenLlamaForCausalLM.from_pretrained(model_name)
 openllama_pipeline = pipeline(
+    "GenerationMixin",
     model=model,
     tokenizer=tokenizer,
     truncation=True,  # Add this line