Jesse Karmani commited on
Commit
59dc238
1 Parent(s): 24fe65c

Add model name input to gradio interface

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -71,8 +71,8 @@ def main(
71
  input_string: str,
72
  source_lang: Optional[str],
73
  target_lang: Optional[str],
74
- starting_batch_size: int = 8,
75
  model_name: str = "facebook/m2m100_1.2B",
 
76
  lora_weights_name_or_path: str = None,
77
  force_auto_device_map: bool = False,
78
  precision: str = None,
@@ -539,5 +539,5 @@ def main(
539
  # trust_remote_code=args.trust_remote_code,
540
  # )
541
 
542
- demo = gradio.Interface(fn=main, inputs=["textbox", "textbox", "textbox"], outputs="textbox")
543
- demo.launch(share=True)
 
71
  input_string: str,
72
  source_lang: Optional[str],
73
  target_lang: Optional[str],
 
74
  model_name: str = "facebook/m2m100_1.2B",
75
+ starting_batch_size: int = 8,
76
  lora_weights_name_or_path: str = None,
77
  force_auto_device_map: bool = False,
78
  precision: str = None,
 
539
  # trust_remote_code=args.trust_remote_code,
540
  # )
541
 
542
+ demo = gradio.Interface(fn=main, inputs=["textbox", "textbox", "textbox", "textbox"], outputs="textbox")
543
+ demo.launch()