Spaces:
Running
Running
added model path
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 61 |
quantization_config=quantization_config,
|
| 62 |
config=config
|
| 63 |
)
|
| 64 |
-
|
| 65 |
"text-generation",
|
| 66 |
model=model,
|
| 67 |
tokenizer=tokenizer,
|
|
|
|
| 61 |
quantization_config=quantization_config,
|
| 62 |
config=config
|
| 63 |
)
|
| 64 |
+
pipeline = transformers.pipeline(
|
| 65 |
"text-generation",
|
| 66 |
model=model,
|
| 67 |
tokenizer=tokenizer,
|