Update main.py

main.py

@@ -9,16 +9,9 @@ import torch
 
 app = FastAPI()
 
-model_id = "meta-llama/Meta-Llama-3-8B"
 
-try:
-    pipeline = transformers.pipeline(
-    "text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto")
 
-    print(pipeline("Hey how are you doing today?"))
-
-except:
-    print("ERROR")
+summarizer = transformers.pipeline("summarization", model="Falconsai/text_summarization")