grv805 commited on
Commit
abb8b3f
·
verified ·
1 Parent(s): d3e0743

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ app=FastAPI()
7
  # Initialize the text generation pipeline
8
  #pipe = pipeline("text2text-generation", model="google/flan-t5-small")
9
  #pipe = pipeline("text-generation", model="openai-community/gpt2")
10
- pipe=pipeline("text-generation", model="google/gemma-2-2b")
11
 
12
  @app.get("/")
13
  def home():
 
7
  # Initialize the text generation pipeline
8
  #pipe = pipeline("text2text-generation", model="google/flan-t5-small")
9
  #pipe = pipeline("text-generation", model="openai-community/gpt2")
10
+ pipe=pipeline("text-generation", model="HuggingFaceTB/SmolLM2-1.7B-Instruct")
11
 
12
  @app.get("/")
13
  def home():