Spaces:

doubledsbv
/

gemma_2_2B_kafka_experimental

Sleeping

doubledsbv commited on Apr 24

Commit

aad79ca

•

1 Parent(s): 0eb0e45

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat1
 tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
-generate_text = transformers.pipeline(
     model=model, tokenizer=tokenizer,
     return_full_text=True,
     task='text-generation',

 tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
+pipeline = transformers.pipeline(
     model=model, tokenizer=tokenizer,
     return_full_text=True,
     task='text-generation',