shorecode commited on
Commit
16f51b6
·
verified ·
1 Parent(s): 2b34323

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -2,11 +2,13 @@ import gradio as gr
2
  from optimum.onnxruntime import ORTModelForSeq2SeqLM
3
  from transformers import pipeline
4
  import logging
 
5
 
6
  logger = logging.Logger('sc_summ')
7
  logger.setLevel(logging.INFO)
8
  # File handler (for logging to file)
9
- file_handler = logging.FileHandler("sc_summ.log", encoding='utf-8')
 
10
  file_handler.setLevel(logging.INFO)
11
  file_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
12
  file_handler.setFormatter(file_formatter)
@@ -37,7 +39,7 @@ def summarize(src):
37
  return "An error occurred"
38
 
39
  def greet(name):
40
- return "Hello " + name + "!!"
41
 
42
  demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
43
  demo.launch()
 
2
  from optimum.onnxruntime import ORTModelForSeq2SeqLM
3
  from transformers import pipeline
4
  import logging
5
+ import os
6
 
7
  logger = logging.Logger('sc_summ')
8
  logger.setLevel(logging.INFO)
9
  # File handler (for logging to file)
10
+ local_dir = os.path.dirname(__file__)
11
+ file_handler = logging.FileHandler(os.path.join(local_dir, "sc_summ.log"), encoding='utf-8')
12
  file_handler.setLevel(logging.INFO)
13
  file_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
14
  file_handler.setFormatter(file_formatter)
 
39
  return "An error occurred"
40
 
41
  def greet(name):
42
+ return "Hello " + __file__ + "!!"
43
 
44
  demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
45
  demo.launch()