Hugging Face's logo

Spaces:
ch404
/
heisterverse
Sleeping

App Files Community
ch404 commited on
Commit
00eea7f
·
1 Parent(s): 6c546fd

Test

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
6
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
7
  import uvicorn
8
- from models import ChatRequest, ChatResponse
9
  from contextlib import asynccontextmanager
10
  import gc
11
 
 
5
  from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
6
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
7
  import uvicorn
8
+ from .models import ChatRequest, ChatResponse
9
  from contextlib import asynccontextmanager
10
  import gc
11