Spaces:
Sleeping
Sleeping
saifeddinemk
commited on
Commit
•
e91e514
1
Parent(s):
71a0d39
Init app
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ from fastapi import FastAPI, HTTPException
|
|
2 |
from pydantic import BaseModel
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
4 |
from peft import PeftModel, PeftConfig
|
5 |
-
|
6 |
# Initialize the FastAPI app
|
7 |
app = FastAPI()
|
8 |
|
|
|
2 |
from pydantic import BaseModel
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
4 |
from peft import PeftModel, PeftConfig
|
5 |
+
import torch
|
6 |
# Initialize the FastAPI app
|
7 |
app = FastAPI()
|
8 |
|