Hjgugugjhuhjggg
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -110,7 +110,7 @@ async def predict(request: DownloadModelRequest):
|
|
110 |
with open(model_path, 'wb') as f:
|
111 |
f.write(model_stream.read())
|
112 |
|
113 |
-
model = AutoModelForCausalLM.from_pretrained(tmp_dir, from_tf=
|
114 |
tokenizer = AutoTokenizer.from_pretrained(tmp_dir)
|
115 |
|
116 |
pipeline_ = pipeline(request.pipeline_task, model=model, tokenizer=tokenizer)
|
|
|
110 |
with open(model_path, 'wb') as f:
|
111 |
f.write(model_stream.read())
|
112 |
|
113 |
+
model = AutoModelForCausalLM.from_pretrained(tmp_dir, from_tf=True)
|
114 |
tokenizer = AutoTokenizer.from_pretrained(tmp_dir)
|
115 |
|
116 |
pipeline_ = pipeline(request.pipeline_task, model=model, tokenizer=tokenizer)
|