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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,8 +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("text2text-generation",
13
- model="mat27/medmnistPrueba")
14
 
15
  # Define a function to handle the GET request at `/generate`
16
  # 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(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