Spaces:

ejschwartz
/

aidapal-space

Running on Zero

ejschwartz commited on 18 days ago

Commit

4f80c95

1 Parent(s): df6ba08

pipeline again

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 # Then create the pipeline with the model and tokenizer
-#pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
 @spaces.GPU
 def greet(name):

 )
 # Then create the pipeline with the model and tokenizer
+pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
 @spaces.GPU
 def greet(name):