removendo chat para teste

Dockerfile

@@ -21,8 +21,7 @@ VOLUME /code/models
 COPY . .
 RUN chmod -R 777 /code
 
-
-RUN transformers-cli download csebuetnlp/mT5_multilingual_XLSum
+ 
 RUN transformers-cli download facebook/m2m100_1.2B
 
 

main.py

@@ -6,7 +6,7 @@ from typing import List, Literal
 from fastapi import FastAPI
 from pydantic import BaseModel
 from enum import Enum
-from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import M2M100Tokenizer, M2M100ForConditionalGeneration
 import torch
 import uvicorn
 app = FastAPI(docs_url="/")
@@ -59,41 +59,7 @@ async def translate(request: TranslationRequest):
 
 
 
-# chat
-WHITESPACE_HANDLER = lambda k: re.sub('\s+', ' ', re.sub('\n+', ' ', k.strip()))
  
-chat_model_name = "csebuetnlp/mT5_multilingual_XLSum"
-tokenizer = AutoTokenizer.from_pretrained(chat_model_name)
-modelchat = AutoModelForSeq2SeqLM.from_pretrained(chat_model_name)
-
-@app.get("/chat")
-async def read_root(text: str):
-    input_ids = tokenizer(
-        [WHITESPACE_HANDLER(text)],
-        return_tensors="pt",
-        padding="max_length",
-        truncation=True,
-        max_length=512
-    )["input_ids"]
-
-        # max_length=84,
-    output_ids = modelchat.generate(
-        input_ids=input_ids,
-        max_length=500,
-        no_repeat_ngram_size=2,
-        num_beams=4
-    )[0]
-
-    summary = tokenizer.decode(
-        output_ids,
-        skip_special_tokens=True,
-        clean_up_tokenization_spaces=True
-    )
-
-    return {"summary": summary}
-
-
-