juancamval commited on
Commit
353ee60
·
verified ·
1 Parent(s): 2f9fd39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -3,6 +3,19 @@ import streamlit as st
3
  import re
4
  from transformers import pipeline
5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  # Pendiente las Keys, dependiendo del modelo que escojamos
7
 
8
  model_name = "google/flan-t5-small" # Probando modelos
 
3
  import re
4
  from transformers import pipeline
5
 
6
+ # funcion para extraccion de codigo del modelo
7
+
8
+ def extract_code(llm_output):
9
+
10
+ code_match = re.search(r"```python\n(.*?)\n```", llm_output, re.DOTALL)
11
+ if code_match:
12
+ return code_match.group(1)
13
+ return None
14
+
15
+ # funcion para prompts
16
+
17
+ # conexion a supabase
18
+
19
  # Pendiente las Keys, dependiendo del modelo que escojamos
20
 
21
  model_name = "google/flan-t5-small" # Probando modelos