MekkCyber commited on
Commit
740d91a
·
1 Parent(s): 637d627
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -5,10 +5,7 @@ import time
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  import logging
7
  from starlette.middleware.sessions import SessionMiddleware
8
- from fastapi import FastAPI
9
- import uvicorn
10
- app = FastAPI()
11
- app.add_middleware(SessionMiddleware, secret_key="secure_key")
12
 
13
  # Configure logging
14
  logging.basicConfig(level=logging.INFO)
@@ -140,9 +137,12 @@ def interface():
140
  return demo
141
 
142
  demo = interface()
143
- app.mount("/", demo)
144
-
145
- # Launch the app
146
- # demo.launch()
147
 
148
- uvicorn.run(app, host="0.0.0.0", port=7860)
 
 
 
 
 
 
 
 
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  import logging
7
  from starlette.middleware.sessions import SessionMiddleware
8
+
 
 
 
9
 
10
  # Configure logging
11
  logging.basicConfig(level=logging.INFO)
 
137
  return demo
138
 
139
  demo = interface()
 
 
 
 
140
 
141
+ # Access FastAPI app instance from Gradio
142
+ fastapi_app = demo.app
143
+
144
+ # Add SessionMiddleware to enable session management
145
+ fastapi_app.add_middleware(SessionMiddleware, secret_key="secret_key") # Use a secure, random secret key
146
+
147
+ # Launch the app
148
+ demo.launch()