Hjgugugjhuhjggg commited on
Commit
606b557
·
verified ·
1 Parent(s): 1836b57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from fastapi import FastAPI, HTTPException
4
  from fastapi.responses import StreamingResponse
5
  from pydantic import BaseModel
6
  from transformers import (
 
7
  AutoModelForCausalLM,
8
  AutoTokenizer,
9
  GenerationConfig,
 
4
  from fastapi.responses import StreamingResponse
5
  from pydantic import BaseModel
6
  from transformers import (
7
+ AutoConfig,
8
  AutoModelForCausalLM,
9
  AutoTokenizer,
10
  GenerationConfig,