RiteshAkhade commited on
Commit
6d5ef45
Β·
verified Β·
1 Parent(s): f089946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -9
app.py CHANGED
@@ -1,21 +1,24 @@
1
- import gradio
2
- import gradio.components as components
3
- import gradio.blocks as blocks
4
 
5
- # Monkey patch for Pydantic boolean schema issue
6
- import gradio as gr
7
- import gradio._types as _types
8
 
9
- if hasattr(_types, "_GCU"):
10
- _types._GCU.get_type = lambda x: bool if x == "boolean" else _types._GCU.get_type(x)
11
 
 
 
 
 
 
 
 
12
  import whisper
13
  import gradio as gr
14
  import torch
15
  from transformers import BertTokenizer, BertForSequenceClassification, pipeline
16
  from app.questions import get_question
17
 
18
- # Load modelswhisper_model = whisper.load_model("small")
 
19
  confidence_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/final_confidence')
20
  confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/final_confidence')
21
  context_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/context_model')
 
 
 
 
1
 
2
+ import gradio_client.utils as _gcu
 
 
3
 
4
+ _original_get_type = _gcu.get_type
5
+ def _get_type_patched(schema):
6
 
7
+ if isinstance(schema, bool):
8
+ return "Any"
9
+ return _original_get_type(schema)
10
+
11
+ _gcu.get_type = _get_type_patched
12
+ # ─────────────────────────────────────────────────────────────────────────────
13
+
14
  import whisper
15
  import gradio as gr
16
  import torch
17
  from transformers import BertTokenizer, BertForSequenceClassification, pipeline
18
  from app.questions import get_question
19
 
20
+ # Load models
21
+ whisper_model = whisper.load_model("small")
22
  confidence_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/final_confidence')
23
  confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/final_confidence')
24
  context_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/context_model')