OnlyCheeini commited on
Commit
bb9d04d
·
verified ·
1 Parent(s): 78f2093

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -3,7 +3,13 @@ from fastapi import FastAPI, HTTPException
3
  from pydantic import BaseModel
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
  import torch
 
6
 
 
 
 
 
 
7
  # Disable hf_transfer
8
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "false"
9
 
 
3
  from pydantic import BaseModel
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
  import torch
6
+ import gradio as gr
7
 
8
+ def greet(name, req: gr.Request):
9
+ return f"{req.headers=}"
10
+
11
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
12
+ iface.launch()
13
  # Disable hf_transfer
14
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "false"
15