Hjgugugjhuhjggg commited on
Commit
9125939
·
verified ·
1 Parent(s): 8f8c61d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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"s3://{self.bucket_name}/{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,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=7860)
 
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)