StefanG2002 commited on
Commit
ad37ea9
·
verified ·
1 Parent(s): d6c82f4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -6,7 +6,7 @@ import uvicorn
6
 
7
  app = FastAPI()
8
 
9
- client = InferenceClient("google/codegemma-7b-it")
10
 
11
  class Item(BaseModel):
12
  prompt: str
 
6
 
7
  app = FastAPI()
8
 
9
+ client = InferenceClient("google/gemma-7b-it")
10
 
11
  class Item(BaseModel):
12
  prompt: str