Spaces:
Sleeping
Sleeping
Hjgugugjhuhjggg
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -108,7 +108,7 @@ class S3DirectStream:
|
|
108 |
tokenizer_stream = self.stream_from_s3(f"{profile}/{model}/tokenizer.json")
|
109 |
tokenizer_data = tokenizer_stream.read().decode("utf-8")
|
110 |
|
111 |
-
tokenizer = AutoTokenizer.from_pretrained(f"
|
112 |
return tokenizer
|
113 |
except Exception as e:
|
114 |
raise HTTPException(status_code=500, detail=f"Error al cargar el tokenizer desde S3: {e}")
|
@@ -184,4 +184,4 @@ async def predict(model_request: dict):
|
|
184 |
|
185 |
if __name__ == "__main__":
|
186 |
import uvicorn
|
187 |
-
uvicorn.run(app, host="0.0.0.0", port=
|
|
|
108 |
tokenizer_stream = self.stream_from_s3(f"{profile}/{model}/tokenizer.json")
|
109 |
tokenizer_data = tokenizer_stream.read().decode("utf-8")
|
110 |
|
111 |
+
tokenizer = AutoTokenizer.from_pretrained(f"{profile}/{model}")
|
112 |
return tokenizer
|
113 |
except Exception as e:
|
114 |
raise HTTPException(status_code=500, detail=f"Error al cargar el tokenizer desde S3: {e}")
|
|
|
184 |
|
185 |
if __name__ == "__main__":
|
186 |
import uvicorn
|
187 |
+
uvicorn.run(app, host="0.0.0.0", port=8000)
|