dohael commited on
Commit
413ffb9
·
verified ·
1 Parent(s): d95047f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  from fastapi import FastAPI
 
2
  from pydantic import BaseModel
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
  from peft import PeftModel
@@ -6,6 +7,14 @@ import torch
6
 
7
  app = FastAPI(title="Assistant IA Education Marocaine")
8
 
 
 
 
 
 
 
 
 
9
  MODEL_NAME = "unsloth/mistral-7b-v0.3"
10
  LORA_NAME = "dohael/mistral-7b-education-maroc"
11
 
 
1
  from fastapi import FastAPI
2
+ from fastapi.middleware.cors import CORSMiddleware
3
  from pydantic import BaseModel
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
  from peft import PeftModel
 
7
 
8
  app = FastAPI(title="Assistant IA Education Marocaine")
9
 
10
+ # Autoriser toutes les origines (CORS)
11
+ app.add_middleware(
12
+ CORSMiddleware,
13
+ allow_origins=["*"],
14
+ allow_methods=["*"],
15
+ allow_headers=["*"],
16
+ )
17
+
18
  MODEL_NAME = "unsloth/mistral-7b-v0.3"
19
  LORA_NAME = "dohael/mistral-7b-education-maroc"
20