mat27 commited on
Commit
5da7bf9
·
1 Parent(s): d2cf07e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ app = FastAPI()
9
  # Initialize the text generation pipeline
10
  # This function will be able to generate text
11
  # given an input.
12
- pipe = pipeline(model="mat27/medmnistPrueba")
13
 
14
  # Define a function to handle the GET request at `/generate`
15
  # The generate() function is defined as a FastAPI route that takes a
 
9
  # Initialize the text generation pipeline
10
  # This function will be able to generate text
11
  # given an input.
12
+ pipe = pipeline("text2text-generation", model="mat27/medmnistPrueba")
13
 
14
  # Define a function to handle the GET request at `/generate`
15
  # The generate() function is defined as a FastAPI route that takes a